Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dreduce.cc72 int3 GetMaximumPossibleWGSize(const std::vector<int>& ordered_sizes, in GetMaximumPossibleWGSize() argument
76 for (int i = ordered_sizes.size() - 1; i >= 0; i--) { in GetMaximumPossibleWGSize()
77 const int wg_index = ordered_sizes.size() - 1 - i; in GetMaximumPossibleWGSize()
81 while (ordered_sizes[i] >= wg_size[wg_index] * 2) { in GetMaximumPossibleWGSize()
116 std::vector<int> ordered_sizes; in Reduce() local
126 ordered_sizes.push_back(reduction_size); in Reduce()
131 GetMaximumPossibleWGSize(ordered_sizes, max_total_wg_size); in Reduce()