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):
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
You can’t perform that action at this time.
0 commit comments