Searched refs:x_alignment (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | workgroup_selection.cc | 33 WorkGroupSizeAlignment x_alignment, in AddCornerCases() argument
48 if (x_alignment == WorkGroupSizeAlignment::PRECISE && in AddCornerCases()
74 if (x_alignment == WorkGroupSizeAlignment::PRECISE && grid.x % x != 0) { in AddCornerCases()
148 const T& max_work_group_sizes, WorkGroupSizeAlignment x_alignment, in GenerateWorkGroupSizes() argument
153 std::vector<int> sizes_x = GetPossibleSizes(grid.x, x_alignment); in GenerateWorkGroupSizes()
180 WorkGroupSizeAlignment x_alignment, WorkGroupSizeAlignment y_alignment,
186 WorkGroupSizeAlignment x_alignment, WorkGroupSizeAlignment y_alignment,
|
| D | workgroup_selection.h | 38 const T& max_work_group_sizes, WorkGroupSizeAlignment x_alignment,
|