未验证 提交 5ec362f2 编写于 作者: C ceci3 提交者: GitHub

fix matmul_v2 (#896)

上级 b3d82cb5
......@@ -21,18 +21,18 @@ __all__ = ['check_search_space']
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'
]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册