You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by ha...@apache.org on 2019/06/24 21:51:52 UTC
[incubator-mxnet] 01/01: [AMP] Move topk from FP16_FP32_FUNCS to
FP32_FUNCS
This is an automated email from the ASF dual-hosted git repository.
haibin pushed a commit to branch amp-patch
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
commit 6052036e8a4b37d91d8ce9b4a5ef059a021fab92
Author: Haibin Lin <li...@gmail.com>
AuthorDate: Mon Jun 24 14:51:01 2019 -0700
[AMP] Move topk from FP16_FP32_FUNCS to FP32_FUNCS
---
python/mxnet/contrib/amp/lists/symbol.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/mxnet/contrib/amp/lists/symbol.py b/python/mxnet/contrib/amp/lists/symbol.py
index 9c99340..066618b 100644
--- a/python/mxnet/contrib/amp/lists/symbol.py
+++ b/python/mxnet/contrib/amp/lists/symbol.py
@@ -340,7 +340,6 @@ FP16_FP32_FUNCS = [
'take',
'tanh',
'tile',
- 'topk',
'transpose',
'trunc',
'uniform',
@@ -463,6 +462,7 @@ FP32_FUNCS = [
'_sparse_norm',
'_sparse_rsqrt',
'argsort',
+ 'topk',
# Neural network
'SoftmaxOutput',