Searched refs:rowSize (Results 1 – 2 of 2) sorted by relevance
32 const int rowSize = inputShape.dimensions.back(); in evalGeneric() local34 std::vector<std::pair<T, int32_t>> values(rowSize); in evalGeneric()37 for (int rowBegin = 0; rowBegin < totalSize; rowBegin += rowSize) { in evalGeneric()38 for (int i = 0; i < rowSize; ++i) { in evalGeneric()41 std::nth_element(values.begin(), values.begin() + (rowSize - k), values.end()); in evalGeneric()42 std::sort(values.begin() + (rowSize - k), values.end()); in evalGeneric()
622 size_t rowSize = mRowStride; in writeToStream() local625 ssize_t bytesRead = mInput->read(rowBytes, rowFillAmt, rowSize); in writeToStream()643 rowSize -= bytesRead; in writeToStream()