Move ops which don't support FP16 dtype to FP32 list

apache · Oct 29, 2019 · a427a5e · a427a5e
1 parent 60d74bc
commit a427a5e
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/python/mxnet/contrib/amp/lists/symbol.py b/python/mxnet/contrib/amp/lists/symbol.py
@@ -32,7 +32,6 @@
 # are dtype neutral (can work in both fp16 and fp32)
 FP16_FP32_FUNCS = [
     'BatchNorm',
-    'BatchNorm_v1',
     'BilinearSampler',
     'BlockGrad',
     'Cast',
@@ -47,7 +46,6 @@
     'GridGenerator',
     'Pad',
     'Pooling',
-    'Pooling_v1',
     'ROIPooling',
     'Reshape',
     'SequenceLast',
@@ -368,6 +366,11 @@
     'arctanh',
     '_sparse_arcsin',
     '_sparse_arctanh',
+    'BatchNorm_v1',
+    'Pooling_v1',
+    '_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',