From aff706fa0b6dd4021390c17aedfd8a7283a3925d Mon Sep 17 00:00:00 2001 From: David Fan Date: Thu, 27 Aug 2020 14:55:06 -0700 Subject: [PATCH] Fix merge --- onnxconverter_common/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxconverter_common/optimizer.py b/onnxconverter_common/optimizer.py index b60d97e..2385aff 100644 --- 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