Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Change to use MXNet's topk for CPUs in inference
Since MXNet's topk has better performance than numpy version with PR apache/mxnet#12085, in order to leverage such performance boost, change to use MXNet's topk for CPU device when doing inference.
- Loading branch information