File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -130,15 +130,15 @@ def apply(self, model):
130
130
outp = post_model .graph .output [0 ]
131
131
132
132
vi_pre = [x for x in pre_model .graph .value_info ]
133
- out_pre = [x for x in pre_model .graph .output ]
134
133
qa_pre = [x for x in pre_model .graph .quantization_annotation ]
135
134
init_pre = [x for x in pre_model .graph .initializer ]
136
135
137
136
vi_post = [x for x in post_model .graph .value_info ]
137
+ in_post = [x for x in post_model .graph .input ]
138
138
qa_post = [x for x in post_model .graph .quantization_annotation ]
139
139
init_post = [x for x in post_model .graph .initializer ]
140
140
141
- vi_new = vi_pre + vi_post + out_pre
141
+ vi_new = vi_pre + vi_post + in_post
142
142
qa_new = qa_pre + qa_post
143
143
init_new = init_pre + init_post
144
144
You can’t perform that action at this time.
0 commit comments