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

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

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