fix quantization of EmbeddingLayerNorm (#5321)

microsoft · Sep 29, 2020 · 5de47af · 5de47af
1 parent c00e13a
commit 5de47af
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/onnxruntime/python/tools/quantization/operators/embed_layernorm.py b/onnxruntime/python/tools/quantization/operators/embed_layernorm.py
@@ -17,6 +17,5 @@ def quantize(self):
         (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