Searched refs:WorkGroupSizeAlignment (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | workgroup_selection.cc | 33 WorkGroupSizeAlignment x_alignment, in AddCornerCases()
34 WorkGroupSizeAlignment y_alignment, in AddCornerCases()
35 WorkGroupSizeAlignment z_alignment, in AddCornerCases()
48 if (x_alignment == WorkGroupSizeAlignment::PRECISE && in AddCornerCases()
52 if (y_alignment == WorkGroupSizeAlignment::PRECISE && in AddCornerCases()
56 if (z_alignment == WorkGroupSizeAlignment::PRECISE && in AddCornerCases()
74 if (x_alignment == WorkGroupSizeAlignment::PRECISE && grid.x % x != 0) { in AddCornerCases()
77 if (y_alignment == WorkGroupSizeAlignment::PRECISE && grid.y % y != 0) { in AddCornerCases()
80 if (z_alignment == WorkGroupSizeAlignment::PRECISE && grid.z % z != 0) { in AddCornerCases()
130 WorkGroupSizeAlignment z_alignment) { in GetPossibleSizes()
[all …]
|
| D | workgroup_selection.h | 28 enum class WorkGroupSizeAlignment { PRECISE, NO_ALIGNMENT }; enum
31 WorkGroupSizeAlignment z_alignment);
38 const T& max_work_group_sizes, WorkGroupSizeAlignment x_alignment,
39 WorkGroupSizeAlignment y_alignment, WorkGroupSizeAlignment z_alignment);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | work_group_picking.h | 34 WorkGroupSizeAlignment z_alignment,
40 WorkGroupSizeAlignment z_alignment,
|
| D | work_group_picking.cc | 37 const GpuInfo& gpu_info, WorkGroupSizeAlignment z_alignment) { in GenerateWorkGroupSizesXYMultipleOf()
67 const GpuInfo& gpu_info, WorkGroupSizeAlignment z_alignment) { in GenerateWorkGroupSizesXMultipleOf()
73 GetPossibleSizes(grid.y, WorkGroupSizeAlignment::PRECISE); in GenerateWorkGroupSizesXMultipleOf()
240 WorkGroupSizeAlignment z_alignment, in GetPossibleWorkGroupsXYMultipleOf()
249 WorkGroupSizeAlignment z_alignment, in GetPossibleWorkGroupsXMultipleOf()
|