saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-01 23:30:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix typo. (#7690)

Browse source
This commit is contained in:
Zhang Lei 2021-05-14 10:25:34 -07:00 committed by GitHub
parent a9b47ca8e7
commit 033f0b3b7c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
onnxruntime/python/tools/quantization/onnx_model.py
View file

@ -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)
self.graph().node.extend(sorted_nodes)