Commit Graph

12 Commits (c21a979790aebefffdde5c470dc406a2a81959e7)

Author SHA1 Message Date
Jiawei Wang 8f035fb637
Add TopK Op Grad CPU&GPU Kernel test=develop (#22628)
5 years ago
zhaoyuchen2018 832d8191ff
Fix topk cannot handle 1D vector bug (#18466)
6 years ago
whs 2314f2ebb3
Make topk op support variable k. (#15044)
6 years ago
qingqing01 c7379a7320 Fix top_k op (#14034)
6 years ago
Sylwester Fraczek 4d55aca40e reserve vector space before loop in top-k
7 years ago
mozga-intel 278f55bcef Support for mklml version of code
7 years ago
mozga-intel e7f0696513 Top K with omp
7 years ago
qingqing01 ee9832a7a9
Add Top-k Python API. (#9973)
7 years ago
Abhinav Arora 981d7d01f3 Fix CPPLint issues in spp_op, sum_op, topk_op, transpose_op, unpool_op and warpctc_op
7 years ago
qingqing01 24509f4af9 Fix the grammar in copyright. (#8403)
7 years ago
Yi Wang fc374821dd Correct #include path
7 years ago
Yi Wang 90648f336d Move file to fluid/; Edit CMakeLists.txt
7 years ago