diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py index ec215a3e5757e..320c14d4e9ca4 100644 --- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py +++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py @@ -81,6 +81,7 @@ "transpose", "pad2d", "reshape", + "layer_norm", ] # list op real input and output names, to avoid processing input such as AxisTensor.