Fix MergeOptimizer for the case Transpose + xxx + Transpose (#142)

microsoft · Aug 27, 2020 · 991ae8a · 991ae8a
1 parent b3e0631
commit 991ae8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxconverter_common/optimizer.py b/onnxconverter_common/optimizer.py
@@ -506,7 +506,7 @@ def apply(self, node_list):
             node_list = self.delete_node_1ton(node_list, self.begin, self.begin_n, self.begin_n.successor[0])
             node_list = self.delete_node_1ton(node_list, self.end_p.get_precedence_by_idx(0), self.end_p, self.end)
         else:
-            node_list = self.delete_node_1ton(node_list, self.begin_n, self.end_p, self.end)
+            node_list = self.delete_node_1ton(node_list, self.end_p.get_precedence_by_idx(0), self.end_p, self.end)
             self.begin_n.origin = helper.make_node('Transpose', self.begin_n.origin.input, self.begin_n.origin.output,
                                                    self.begin_n.origin.name, perm=perm_f)
         return node_list, True