reserve vector space before loop in top-k

guochaorong-patch-1
Sylwester Fraczek 7 years ago
parent 76086df412
commit 4d55aca40e

@ -60,6 +60,7 @@ class TopkKernel : public framework::OpKernel<T> {
#endif
for (size_t i = 0; i < row; i++) {
std::vector<std::pair<T, size_t>> vec;
vec.reserve(col);
for (size_t j = 0; j < col; j++) {
vec.push_back(std::pair<T, size_t>(eg_input(i, j), j));
}

Loading…
Cancel
Save