From 88e9d74a75dc0add21c0b8f4e372e53f2bccd0f9 Mon Sep 17 00:00:00 2001 From: cc <52520497+juncaipeng@users.noreply.github.com> Date: Tue, 26 May 2020 15:50:00 +0800 Subject: [PATCH] Collecting concat output threshold, test=develop (#24742) --- .../paddle/fluid/contrib/slim/quantization/quantization_pass.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py index 44a872bd76..fcf7a51113 100644 --- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py +++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py @@ -43,7 +43,7 @@ _fake_quant_dequant_op_list = [ _out_scale_op_list = [ "conv2d", "depthwise_conv2d", "mul", "matmul", "relu", "leaky_relu", "relu6", "sigmoid", "tanh", "prelu", "swish", "softmax", "batch_norm", - "elementwise_add", "pool2d", "reshape2", "transpose2" + "elementwise_add", "pool2d", "reshape2", "transpose2", "concat" ] # list op real input and output names, to avoid processing input such as AxisTensor. -- GitLab