Skip to content

Commit 7380e62

Browse files
authored
sync optimizer from keras-onnx (#224)
1 parent 8e9f1d1 commit 7380e62

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onnxutils/onnxtk/optimizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -363,10 +363,10 @@ def apply(self, node_list):
363363
['fan_in_adjustment_in' + str(FanInSolution.number)],
364364
['fan_in_adjustment_out' + str(FanInSolution.number)],
365365
perm=self.perm,
366-
name='TransposeFanIn' + str(FanOutSolution.number)))
366+
name='TransposeFanIn' + str(FanInSolution.number)))
367367
FanInSolution.number = FanInSolution.number + 1
368368
precedence_list = self.begin.precedence.copy()
369-
node_list = Solution.add_siso_node(node_list, self.begin, self.begin_n, list(self.begin_n.input.values())[0], nnode)
369+
node_list = Solution.add_siso_node(node_list, self.begin, self.begin_n, list(self.begin.output.values())[0], nnode)
370370
for branch in precedence_list:
371371
node_list = Solution.delete_node(node_list, branch.precedence[0], branch, self.begin)
372372
return node_list

0 commit comments

Comments
 (0)