fix flatten in quant (#37741)

a4c0c71c · Guanghua Yu · GitHub · 026de65c · a4c0c71c
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/contrib/slim/quantization/quantization_pass.py ...ddle/fluid/contrib/slim/quantization/quantization_pass.py +1 -1

未找到文件。
--- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
+++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
@@ -136,7 +136,7 @@ _op_real_in_out_name = {
    "flatten": [["X"], ["Out"]],
    "flatten2": [["X"], ["Out"]],
    "unsqueeze2": [["X"], ["Out"]],
-    "flatten_contiguous_range": [['X'], ["Out", "XShape"]],
+    "flatten_contiguous_range": [['X'], ["Out"]],
 }

 _conv_ops = ['conv2d', 'depthwise_conv2d', 'conv2d_transpose']