Searched refs:max_work_group_total_size (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | workgroup_selection.cc | 31 void AddCornerCases(const T& grid, int max_work_group_total_size, in AddCornerCases() argument
45 wg_x * wg_y * wg_z > max_work_group_total_size) { in AddCornerCases()
71 x * y * z > max_work_group_total_size) { in AddCornerCases()
147 const T& grid, int min_work_group_total_size, int max_work_group_total_size, in GenerateWorkGroupSizes() argument
165 work_group_size > max_work_group_total_size) in GenerateWorkGroupSizes()
179 int max_work_group_total_size, const int3& max_work_group_sizes,
185 int max_work_group_total_size, const uint3& max_work_group_sizes,
192 const int max_work_group_total_size, in GenerateWorkGroupSizesAlignedToGrid() argument
196 grid, /*min_work_group_total_size = */ 32, max_work_group_total_size, in GenerateWorkGroupSizesAlignedToGrid()
200 AddCornerCases(grid, max_work_group_total_size, max_work_group_size, in GenerateWorkGroupSizesAlignedToGrid()
[all …]
|
| D | workgroup_selection.h | 37 const T& grid, int min_work_group_total_size, int max_work_group_total_size,
44 const int max_work_group_total_size,
|
| D | gpu_info.h | 278 int max_work_group_total_size; member
|
| D | gpu_info.cc | 580 return opencl_info.max_work_group_total_size; in GetMaxWorkGroupTotalSize()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_device.cc | 218 info.opencl_info.max_work_group_total_size = in GpuInfoFromDeviceID()
|