We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 279f9c3 commit 07fda66Copy full SHA for 07fda66
src/qonnx/transformation/merge_onnx_models.py
@@ -130,15 +130,15 @@ def apply(self, model):
130
outp = post_model.graph.output[0]
131
132
vi_pre = [x for x in pre_model.graph.value_info]
133
- out_pre = [x for x in pre_model.graph.output]
134
qa_pre = [x for x in pre_model.graph.quantization_annotation]
135
init_pre = [x for x in pre_model.graph.initializer]
136
137
vi_post = [x for x in post_model.graph.value_info]
+ in_post = [x for x in post_model.graph.input]
138
qa_post = [x for x in post_model.graph.quantization_annotation]
139
init_post = [x for x in post_model.graph.initializer]
140
141
- vi_new = vi_pre + vi_post + out_pre
+ vi_new = vi_pre + vi_post + in_post
142
qa_new = qa_pre + qa_post
143
init_new = init_pre + init_post
144
0 commit comments