From 5ec362f273cceadb7d361b1c7afc5e617a398a96 Mon Sep 17 00:00:00 2001 From: ceci3 Date: Wed, 15 Sep 2021 19:26:02 +0800 Subject: [PATCH] fix matmul_v2 (#896) --- paddleslim/nas/ofa/get_sub_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddleslim/nas/ofa/get_sub_model.py b/paddleslim/nas/ofa/get_sub_model.py index 62414852b6af7..64b02d3f3ce15 100644 --- 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' ]