Update symbol.py (#17408)

eric-haibin-lin · eric-haibin-lin · commit 2c61787c78a5 · 2020-01-24T15:11:04.000-08:00
diff --git a/python/mxnet/contrib/amp/lists/symbol.py b/python/mxnet/contrib/amp/lists/symbol.py
@@ -591,6 +591,10 @@
     '_contrib_dgl_graph_compact',
     '_contrib_dgl_subgraph',
     '_contrib_edge_id',
+    '_contrib_interleaved_matmul_encdec_qk',
+    '_contrib_interleaved_matmul_encdec_valatt',
+    '_contrib_interleaved_matmul_selfatt_qk',
+    '_contrib_interleaved_matmul_selfatt_valatt',
     'where',
     '_sparse_where',
     '_sparse_broadcast_add',
