Searched refs:z_alignment (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | workgroup_selection.cc | 35 WorkGroupSizeAlignment z_alignment, in AddCornerCases() argument
56 if (z_alignment == WorkGroupSizeAlignment::PRECISE && in AddCornerCases()
80 if (z_alignment == WorkGroupSizeAlignment::PRECISE && grid.z % z != 0) { in AddCornerCases()
130 WorkGroupSizeAlignment z_alignment) { in GetPossibleSizes() argument
131 if (z_alignment == WorkGroupSizeAlignment::PRECISE) { in GetPossibleSizes()
149 WorkGroupSizeAlignment y_alignment, WorkGroupSizeAlignment z_alignment) { in GenerateWorkGroupSizes() argument
155 std::vector<int> sizes_z = GetPossibleSizes(grid.z, z_alignment); in GenerateWorkGroupSizes()
181 WorkGroupSizeAlignment z_alignment);
187 WorkGroupSizeAlignment z_alignment);
|
| D | workgroup_selection.h | 31 WorkGroupSizeAlignment z_alignment);
39 WorkGroupSizeAlignment y_alignment, WorkGroupSizeAlignment z_alignment);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | work_group_picking.cc | 37 const GpuInfo& gpu_info, WorkGroupSizeAlignment z_alignment) { in GenerateWorkGroupSizesXYMultipleOf() argument
41 std::vector<int> possible_z_sizes = GetPossibleSizes(grid.z, z_alignment); in GenerateWorkGroupSizesXYMultipleOf()
67 const GpuInfo& gpu_info, WorkGroupSizeAlignment z_alignment) { in GenerateWorkGroupSizesXMultipleOf() argument
71 std::vector<int> possible_z_sizes = GetPossibleSizes(grid.z, z_alignment); in GenerateWorkGroupSizesXMultipleOf()
240 WorkGroupSizeAlignment z_alignment, in GetPossibleWorkGroupsXYMultipleOf() argument
243 multiplier, grid, kernel_info, gpu_info, z_alignment); in GetPossibleWorkGroupsXYMultipleOf()
249 WorkGroupSizeAlignment z_alignment, in GetPossibleWorkGroupsXMultipleOf() argument
252 multiplier, grid, kernel_info, gpu_info, z_alignment); in GetPossibleWorkGroupsXMultipleOf()
|
| D | work_group_picking.h | 34 WorkGroupSizeAlignment z_alignment,
40 WorkGroupSizeAlignment z_alignment,
|