diff --git a/python/paddle/fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py b/python/paddle/fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py index a79b1ee18b121711e4dc671c1b5b31f2960d4d2e..d93a2059bdcf080c5bbd5546469f68f800c583f9 100644 --- a/python/paddle/fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py +++ b/python/paddle/fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py @@ -154,8 +154,7 @@ class Quant2Int8MkldnnPass(object): is_quantized_op = True for var_node in op_node.inputs: for front_op_node in var_node.inputs: - if "fake_quantize_dequantize_" not in front_op_node.name( - ): + if "quantize_dequantize" not in front_op_node.name(): is_quantized_op = False if not is_quantized_op: op_node.op()._set_attr("skip_quant", True)