From e8b9aa1f2933f59a6418a208192d1f65b628e05b Mon Sep 17 00:00:00 2001 From: Yufeng Li Date: Tue, 29 Sep 2020 01:00:47 -0700 Subject: [PATCH] fix quantization of EmbeddingLayerNorm (#5321) --- .../python/tools/quantization/operators/embed_layernorm.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/onnxruntime/python/tools/quantization/operators/embed_layernorm.py b/onnxruntime/python/tools/quantization/operators/embed_layernorm.py index b110e67115..6723b3baff 100644 --- a/onnxruntime/python/tools/quantization/operators/embed_layernorm.py +++ b/onnxruntime/python/tools/quantization/operators/embed_layernorm.py @@ -17,6 +17,5 @@ class EmbedLayerNormalizationQuant(QuantOperatorBase): (quantized_input_names, zero_point_names, scale_names, nodes) = \ self.quantizer.quantize_inputs(node, [2, 3, 4]) - nodes.append(node) - - self.quantizer.new_nodes += nodes + super().quantize() + self.quantizer.new_nodes += nodes \ No newline at end of file