Fix amp list.

apache · Feb 4, 2022 · dd01051 · dd01051
1 parent db70227
commit dd01051
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/mxnet/amp/lists/symbol_fp16.py b/python/mxnet/amp/lists/symbol_fp16.py
@@ -108,6 +108,7 @@
     '_contrib_quantized_elemwise_add',
     '_contrib_quantized_act',
     '_contrib_quantized_reshape',
+    '_npx_quantized_reshape',
     '_image_crop',
     '_linspace',
     '_contrib_requantize',