From 09a57a6f0a639278befcd7e7f8dc2ef2418e77cb Mon Sep 17 00:00:00 2001 From: Haibin Lin Date: Thu, 23 Jan 2020 15:36:47 -0800 Subject: [PATCH] Update symbol.py (#17408) --- python/mxnet/contrib/amp/lists/symbol.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/mxnet/contrib/amp/lists/symbol.py b/python/mxnet/contrib/amp/lists/symbol.py index 2146853a6866..d501a7d6c5b5 100644 --- 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',