Add the index_array operator to the AMP symbol list

apache · May 22, 2019 · 69a9969 · 69a9969
1 parent 496bad3
commit 69a9969
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/mxnet/contrib/amp/lists/symbol.py b/python/mxnet/contrib/amp/lists/symbol.py
@@ -95,6 +95,7 @@
     '_contrib_gradientmultiplier',
     '_contrib_group_adagrad_update',
     '_contrib_ifft',
+    '_contrib_index_array',
     '_contrib_index_copy',
     '_contrib_quadratic',
     '_contrib_quantize',