Searched refs:rowSize (Results 1 – 1 of 1) sorted by relevance
36 const int rowSize = inputShape.dimensions.back(); in evalGeneric() local38 std::vector<std::pair<T, int32_t>> values(rowSize); in evalGeneric()41 for (int rowBegin = 0; rowBegin < totalSize; rowBegin += rowSize) { in evalGeneric()42 for (int i = 0; i < rowSize; ++i) { in evalGeneric()45 std::nth_element(values.begin(), values.begin() + (rowSize - k), values.end()); in evalGeneric()46 std::sort(values.begin() + (rowSize - k), values.end()); in evalGeneric()