Skip to content

Commit 07fda66

Browse files
committed
Fix value_info in MergeOnnxModels
1 parent 279f9c3 commit 07fda66

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/qonnx/transformation/merge_onnx_models.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -130,15 +130,15 @@ def apply(self, model):
130130
outp = post_model.graph.output[0]
131131

132132
vi_pre = [x for x in pre_model.graph.value_info]
133-
out_pre = [x for x in pre_model.graph.output]
134133
qa_pre = [x for x in pre_model.graph.quantization_annotation]
135134
init_pre = [x for x in pre_model.graph.initializer]
136135

137136
vi_post = [x for x in post_model.graph.value_info]
137+
in_post = [x for x in post_model.graph.input]
138138
qa_post = [x for x in post_model.graph.quantization_annotation]
139139
init_post = [x for x in post_model.graph.initializer]
140140

141-
vi_new = vi_pre + vi_post + out_pre
141+
vi_new = vi_pre + vi_post + in_post
142142
qa_new = qa_pre + qa_post
143143
init_new = init_pre + init_post
144144

0 commit comments

Comments
 (0)