diff --git a/onnxruntime/python/tools/quantization/operators/lstm.py b/onnxruntime/python/tools/quantization/operators/lstm.py index cd7d296595..3389a9bb63 100644 --- a/onnxruntime/python/tools/quantization/operators/lstm.py +++ b/onnxruntime/python/tools/quantization/operators/lstm.py @@ -99,5 +99,8 @@ class LSTMQuant(QuantOperatorBase): quant_lstm_name = "" if node.name == "" else node.name + "_quant" quant_lstm_node = onnx.helper.make_node("DynamicQuantizeLSTM", inputs, node.output, quant_lstm_name, **kwargs) + self.quantizer.new_nodes.append(quant_lstm_node) - self.quantizer.new_nodes += [quant_lstm_node] + dequantize_node = self.quantizer._dequantize_value(node.input[0]) + if dequantize_node is not None: + self.quantizer.new_nodes.append(dequantize_node)