Skip to content

Commit 24d98be

Browse files
committed
fix loop_rewriter unused output shape
1 parent cec8368 commit 24d98be

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/rewriter/loop_rewriter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def _create_loop_node(self, context, loop_props):
114114
self.g.remove_node(n.name)
115115
else:
116116
loop_outputs.append(utils.make_name("unused_loop_output_"))
117-
loop_output_shapes.append(-1)
117+
loop_output_shapes.append([-1])
118118
loop_output_dtypes.append(None)
119119

120120
# trip count and cond are not used, giving them values just because bug

0 commit comments

Comments
 (0)