Searched refs:wg_z (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | workgroup_selection.cc | 42 int wg_z = DivideRoundUp(grid.z, z); in AddCornerCases() local
44 wg_z > max_work_group_sizes.z || in AddCornerCases()
45 wg_x * wg_y * wg_z > max_work_group_total_size) { in AddCornerCases()
57 grid.z % wg_z != 0) { in AddCornerCases()
60 work_groups->push_back({wg_x, wg_y, wg_z}); in AddCornerCases()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | work_group_picking.cc | 219 int wg_z = GetBiggestDividerWithPriority(grid.z, 8); in GetWorkGroup() local
220 int wg_xy_size = max_size / wg_z; in GetWorkGroup()
223 return int3(wg_x, wg_y, wg_z); in GetWorkGroup()
227 int wg_z = GetBiggestDivider(grid.z, max_z_size); in GetWorkGroupConv() local
228 int wg_xy_size = std::min(256, max_size) / wg_z; in GetWorkGroupConv()
234 return int3(wg_x, wg_y, wg_z); in GetWorkGroupConv()
|
| /external/deqp-deps/amber/tests/cases/ |
| D | shader_specialization.amber | 82 %wg_z = OpCompositeExtract %int %wg_size 2
84 %x_y_z = OpIMul %int %x_y %wg_z
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 206 const std::string wg_z = std::to_string(work_group_size.z); in GetReduceKernelCode() local
304 const std::vector<std::string> local_sizes = {wg_x, wg_y, wg_z}; in GetReduceKernelCode()
|