Home
last modified time | relevance | path

Searched refs:max_work_group_sizes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dworkgroup_selection.cc32 const T& max_work_group_sizes, in AddCornerCases() argument
43 if (wg_x > max_work_group_sizes.x || wg_y > max_work_group_sizes.y || in AddCornerCases()
44 wg_z > max_work_group_sizes.z || in AddCornerCases()
69 if (x > max_work_group_sizes.x || y > max_work_group_sizes.y || in AddCornerCases()
70 z > max_work_group_sizes.z || in AddCornerCases()
148 const T& max_work_group_sizes, WorkGroupSizeAlignment x_alignment, in GenerateWorkGroupSizes() argument
158 if (x > max_work_group_sizes.x) continue; in GenerateWorkGroupSizes()
160 if (y > max_work_group_sizes.y) continue; in GenerateWorkGroupSizes()
162 if (z > max_work_group_sizes.z) continue; in GenerateWorkGroupSizes()
179 int max_work_group_total_size, const int3& max_work_group_sizes,
[all …]
Dworkgroup_selection.h38 const T& max_work_group_sizes, WorkGroupSizeAlignment x_alignment,
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_device.cc213 int3 max_work_group_sizes; in GpuInfoFromDeviceID() local
214 GetDeviceWorkDimsSizes(id, &max_work_group_sizes); in GpuInfoFromDeviceID()
215 info.opencl_info.max_work_group_size_x = max_work_group_sizes.x; in GpuInfoFromDeviceID()
216 info.opencl_info.max_work_group_size_y = max_work_group_sizes.y; in GpuInfoFromDeviceID()
217 info.opencl_info.max_work_group_size_z = max_work_group_sizes.z; in GpuInfoFromDeviceID()