From 5aa74e066277c2b55a8307f32973425bf85de4c0 Mon Sep 17 00:00:00 2001 From: Anirudh Subramanian Date: Tue, 29 Oct 2019 13:30:57 -0700 Subject: [PATCH] Move ops which don't support FP16 dtype to FP32 list (#16668) --- python/mxnet/contrib/amp/lists/symbol.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/mxnet/contrib/amp/lists/symbol.py b/python/mxnet/contrib/amp/lists/symbol.py index 397f4775f8cd..2146853a6866 100644 --- a/python/mxnet/contrib/amp/lists/symbol.py +++ b/python/mxnet/contrib/amp/lists/symbol.py @@ -368,6 +368,9 @@ 'arctanh', '_sparse_arcsin', '_sparse_arctanh', + '_contrib_MultiBoxDetection', + '_contrib_MultiBoxPrior', + '_contrib_MultiBoxTarget', # Exponents 'exp', @@ -575,9 +578,6 @@ 'stack', '_Maximum', '_Minimum', - '_contrib_MultiBoxDetection', - '_contrib_MultiBoxPrior', - '_contrib_MultiBoxTarget', '_contrib_MultiProposal', '_contrib_PSROIPooling', '_contrib_Proposal',