This repository was archived by the owner on Nov 17, 2023. It is now read-only.

[AMP] Move topk from FP16_FP32_FUNCS to FP32_FUNCS #15342

Merged

eric-haibin-lin merged 1 commit into master from amp-patch

Jun 28, 2019

python/mxnet/contrib/amp/lists/symbol.py

-Original file line number
+Diff line change
@@ Expand Up / @@ -340,7 +340,6 @@ @@
         'take',
         'tanh',
         'tile',
-        'topk',
         'transpose',
         'trunc',
         'uniform',
@@ Expand Down Expand Up / @@ -463,6 +462,7 @@ @@
         '_sparse_norm',
         '_sparse_rsqrt',
         'argsort',
+        'topk',
         # Neural network
         'SoftmaxOutput',
@@ Expand Down @@