Searched refs:wg_x (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | workgroup_selection.cc | 40 int wg_x = DivideRoundUp(grid.x, x); 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()
49 grid.x % wg_x != 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 | 221 int wg_x = std::min(DivideRoundUp(grid.x, 2), wg_xy_size); in GetWorkGroup() local
222 int wg_y = std::min(wg_xy_size / wg_x, grid.y); in GetWorkGroup()
223 return int3(wg_x, wg_y, wg_z); in GetWorkGroup()
229 int wg_x = std::min(grid.x, wg_xy_size); in GetWorkGroupConv() local
230 int wg_y = std::min(wg_xy_size / wg_x, grid.y); in GetWorkGroupConv()
234 return int3(wg_x, wg_y, wg_z); in GetWorkGroupConv()
|
| /external/deqp-deps/amber/tests/cases/ |
| D | shader_specialization.amber | 80 %wg_x = OpCompositeExtract %int %wg_size 0
83 %x_y = OpIMul %int %wg_x %wg_y
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 204 const std::string wg_x = std::to_string(work_group_size.x); in GetReduceKernelCode() local
218 c += " int local_id = local_y * " + wg_x + " + local_x;\n"; in GetReduceKernelCode()
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()
|