From 5de47affb14bbb7e0cea35c0f12c6b7f0436025e 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