Searched refs:wg_y (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | workgroup_selection.cc | 41 int wg_y = DivideRoundUp(grid.y, y); in AddCornerCases() local
43 if (wg_x > max_work_group_sizes.x || wg_y > max_work_group_sizes.y || in AddCornerCases()
45 wg_x * wg_y * wg_z > max_work_group_total_size) { in AddCornerCases()
53 grid.y % wg_y != 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 | 222 int wg_y = std::min(wg_xy_size / wg_x, grid.y); in GetWorkGroup() local
223 return int3(wg_x, wg_y, wg_z); in GetWorkGroup()
230 int wg_y = std::min(wg_xy_size / wg_x, grid.y); in GetWorkGroupConv() local
231 if (wg_y == grid.y && grid.y % 2 == 0) { in GetWorkGroupConv()
232 wg_y = grid.y / 2; in GetWorkGroupConv()
234 return int3(wg_x, wg_y, wg_z); in GetWorkGroupConv()
|
| /external/deqp-deps/amber/tests/cases/ |
| D | shader_specialization.amber | 81 %wg_y = OpCompositeExtract %int %wg_size 1
83 %x_y = OpIMul %int %wg_x %wg_y
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 205 const std::string wg_y = std::to_string(work_group_size.y); in GetReduceKernelCode() local
223 c += " int local_id = (local_z * " + wg_y + " + local_y) * " + wg_x + in GetReduceKernelCode()
304 const std::vector<std::string> local_sizes = {wg_x, wg_y, wg_z}; in GetReduceKernelCode()
|