From 033f0b3b7c4b6c1bf70747df3d7ca68c42c9a2d3 Mon Sep 17 00:00:00 2001 From: Zhang Lei Date: Fri, 14 May 2021 10:25:34 -0700 Subject: [PATCH] fix typo. (#7690) --- onnxruntime/python/tools/quantization/onnx_model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/onnxruntime/python/tools/quantization/onnx_model.py b/onnxruntime/python/tools/quantization/onnx_model.py index 510774a11b..0cda0a4a59 100644 --- a/onnxruntime/python/tools/quantization/onnx_model.py +++ b/onnxruntime/python/tools/quantization/onnx_model.py @@ -272,7 +272,7 @@ class ONNXModel: # initialize sorted_nodes sorted_nodes = [] for input in itertools.chain(self.initializer(), self.model.graph.input): - if input_name in deps_to_nodes: + if input.name in deps_to_nodes: for node_idx in deps_to_nodes[input.name]: deps_count[node_idx] = deps_count[node_idx] - 1 if deps_count[node_idx] == 0: @@ -293,4 +293,5 @@ class ONNXModel: assert(e == len(self.graph().node)), "Graph is not a DAG" self.graph().ClearField('node') - self.graph().node.extend(sorted_nodes) \ No newline at end of file + self.graph().node.extend(sorted_nodes) +