fix matmul_v2 (PaddlePaddle#896)

lizexu123 · Sep 15, 2021 · 5ec362f · 5ec362f
1 parent b3d82cb
commit 5ec362f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/paddleslim/nas/ofa/get_sub_model.py b/paddleslim/nas/ofa/get_sub_model.py
@@ -21,18 +21,18 @@
 
 DYNAMIC_WEIGHT_OP = [
     'conv2d', 'mul', 'matmul', 'embedding', 'conv2d_transpose',
-    'depthwise_conv2d'
+    'depthwise_conv2d', 'matmul_v2'
 ]
 
 CONV_TYPES = [
     'conv2d', 'conv3d', 'conv1d', 'superconv2d', 'supergroupconv2d',
-    'superdepthwiseconv2d'
+    'superdepthwiseconv2d', 'matmul_v2'
 ]
 
 ALL_WEIGHT_OP = [
     'conv2d', 'mul', 'matmul', 'elementwise_add', 'embedding',
     'conv2d_transpose', 'depthwise_conv2d', 'batch_norm', 'layer_norm',
-    'instance_norm', 'sync_batch_norm'
+    'instance_norm', 'sync_batch_norm', 'matmul_v2'
 ]