fix matmul_v2 (PaddlePaddle#896)

ceci3 · web-flow · commit 5ec362f273cc · 2021-09-15T19:26:02.000+08:00
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'
 ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -21,18 +21,18 @@`
`21`	`21`
`22`	`22`	`DYNAMIC_WEIGHT_OP = [`
`23`	`23`	`'conv2d', 'mul', 'matmul', 'embedding', 'conv2d_transpose',`
`24`		`- 'depthwise_conv2d'`
	`24`	`+ 'depthwise_conv2d', 'matmul_v2'`
`25`	`25`	`]`
`26`	`26`
`27`	`27`	`CONV_TYPES = [`
`28`	`28`	`'conv2d', 'conv3d', 'conv1d', 'superconv2d', 'supergroupconv2d',`
`29`		`- 'superdepthwiseconv2d'`
	`29`	`+ 'superdepthwiseconv2d', 'matmul_v2'`
`30`	`30`	`]`
`31`	`31`
`32`	`32`	`ALL_WEIGHT_OP = [`
`33`	`33`	`'conv2d', 'mul', 'matmul', 'elementwise_add', 'embedding',`
`34`	`34`	`'conv2d_transpose', 'depthwise_conv2d', 'batch_norm', 'layer_norm',`
`35`		`- 'instance_norm', 'sync_batch_norm'`
	`35`	`+ 'instance_norm', 'sync_batch_norm', 'matmul_v2'`
`36`	`36`	`]`
`37`	`37`
`38`	`38`