diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py index eba881a2637aec97815447dacf04ffd412ca559c..68bf9ecd80be4c0d074508ccf1e8821c230dd709 100644 --- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py +++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py @@ -69,6 +69,10 @@ _out_scale_op_list = [ "hard_swish", "hard_sigmoid", "conv2d_transpose", + "gru", + "bilinear_interp", + "nearest_interp", + "trilinear_interp", ] # list op real input and output names, to avoid processing input such as AxisTensor. @@ -114,6 +118,7 @@ _op_real_in_out_name = { "scale": [["X"], ["Out"]], "hard_swish": [["X"], ["Out"]], "hard_sigmoid": [["X"], ["Out"]], + "gru": [["Input", "Weight"], ["Hidden"]], } _conv_ops = ['conv2d', 'depthwise_conv2d', 'conv2d_transpose']