Pass succ.successor to MergeSolution rather than succ.successor[0] (#120

)
microsoft · Jun 5, 2020 · 0030d9f · 0030d9f
1 parent 598ce54
commit 0030d9f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxconverter_common/optimizer.py b/onnxconverter_common/optimizer.py
@@ -779,7 +779,7 @@ def find(node):
                         else:
                             break
                     if succ.is_transpose:
-                        solution = MergeSolution(node.get_precedence_by_idx(0), node, succ, succ.successor[0])
+                        solution = MergeSolution(node.get_precedence_by_idx(0), node, succ, succ.successor)
                         return solution
 
                 last_switchable = node