|
|
@ -55,6 +55,9 @@ class TopkKernel : public framework::OpKernel<T> {
|
|
|
|
// NOTE: eigen shape doesn't affect paddle tensor.
|
|
|
|
// NOTE: eigen shape doesn't affect paddle tensor.
|
|
|
|
eg_input.reshape(flat2dims);
|
|
|
|
eg_input.reshape(flat2dims);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#ifdef PADDLE_WITH_MKLDNN
|
|
|
|
|
|
|
|
#pragma omp parallel for
|
|
|
|
|
|
|
|
#endif
|
|
|
|
for (size_t i = 0; i < row; i++) {
|
|
|
|
for (size_t i = 0; i < row; i++) {
|
|
|
|
std::vector<std::pair<T, size_t>> vec;
|
|
|
|
std::vector<std::pair<T, size_t>> vec;
|
|
|
|
for (size_t j = 0; j < col; j++) {
|
|
|
|
for (size_t j = 0; j < col; j++) {
|
|
|
|