Home
last modified time | relevance | path

Searched refs:GetPossibleSizes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dworkgroup_selection.cc129 std::vector<int> GetPossibleSizes(int number, in GetPossibleSizes() function
153 std::vector<int> sizes_x = GetPossibleSizes(grid.x, x_alignment); in GenerateWorkGroupSizes()
154 std::vector<int> sizes_y = GetPossibleSizes(grid.y, y_alignment); in GenerateWorkGroupSizes()
155 std::vector<int> sizes_z = GetPossibleSizes(grid.z, z_alignment); in GenerateWorkGroupSizes()
Dworkgroup_selection.h30 std::vector<int> GetPossibleSizes(int number,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dwork_group_picking.cc41 std::vector<int> possible_z_sizes = GetPossibleSizes(grid.z, z_alignment); in GenerateWorkGroupSizesXYMultipleOf()
71 std::vector<int> possible_z_sizes = GetPossibleSizes(grid.z, z_alignment); in GenerateWorkGroupSizesXMultipleOf()
73 GetPossibleSizes(grid.y, WorkGroupSizeAlignment::PRECISE); in GenerateWorkGroupSizesXMultipleOf()