From 478ea78b18f8549fb9d46305392beedd95caa6d1 Mon Sep 17 00:00:00 2001 From: ceci3 Date: Fri, 18 Jun 2021 18:40:21 +0800 Subject: [PATCH] add layernorm (#33610) --- .../paddle/fluid/contrib/slim/quantization/quantization_pass.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py index ec215a3e575..320c14d4e9c 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 @@ _out_scale_op_list = [ "transpose", "pad2d", "reshape", + "layer_norm", ] # list op real input and output names, to avoid processing input such as AxisTensor. -- GitLab