未验证 提交 1f57ac12 编写于 作者: J juncaipeng 提交者: GitHub

delete concat in AddQuantDequantPass, test=develop (#21454)

上级 2a54c359
......@@ -1250,7 +1250,7 @@ class AddQuantDequantPass(object):
moving_rate=0.9,
quant_bits=8,
skip_pattern=["skip_quant"],
quantizable_op_type=["elementwise_add", "pool2d", "concat"],
quantizable_op_type=["elementwise_add", "pool2d"],
is_full_quantized=False):
"""
This pass add quant_dequant op for some ops, of which all the inputs must be
......@@ -1269,7 +1269,7 @@ class AddQuantDequantPass(object):
detected in an op's name scope, the corresponding op will not be quantized.
Default is 'skip_quant'.
quantizable_op_type(list[str], optional): List the type of ops that will be
quantized. Default is ["elementwise_add", "pool2d", "concat"].
quantized. Default is ["elementwise_add", "pool2d"].
is_full_quantized(bool, optional): If set is_full_quantized as True, apply
quantization to all supported quantizable op type. If set is_full_quantized
as False, only apply quantization to the op type according to the input
......@@ -1300,8 +1300,8 @@ class AddQuantDequantPass(object):
def apply(self, graph):
"""
Add quant_dequant before some ops, such as the 'elementwise_add',
'pool2d' and 'concat' op.
Add quant_dequant before some ops, such as the 'elementwise_add' and
'pool2d' op.
Args:
graph(IrGraph): the target graph.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册