Merge pull request #643 from zhijxu-MS/fix_bug

_handle_node_having_branches only handle node which only has one output

Author: zhijxu-MS

tf2onnx/optimizer/transpose_optimizer.py

@@ -197,7 +197,7 @@ def _handle_node_having_branches(self, node):
             return False
         # make sure node's all input transpose all have only 1 consumer node,
         # otherwise, it would impact their other output nodes
-        if self._nodes_has_single_consumer_node(node.inputs):
+        if self._nodes_has_single_consumer_node(node.inputs) and len(node.output) == 1:
             self._create_transpose_pairs_after_node(node)
             input_transposes = set(node.inputs)
             for n in input_transposes: