Skip to content

Commit 2ce73d4

Browse files
committed
fix bug
mul input may come from same transpose node
1 parent b2d3f97 commit 2ce73d4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -380,7 +380,7 @@ def _mul_handler(self, trans, node):
380380
multiplier_input_id = i
381381
multiplier_input_node = input_node
382382

383-
if not multiplier_input_node.is_const():
383+
if multiplier_input_node is None or not multiplier_input_node.is_const():
384384
return False
385385
multiplier = multiplier_input_node.get_tensor_value(as_list=False)
386386

0 commit comments

Comments
 (0)