Skip to content

Commit 0ca1d60

Browse files
committed
remove shape
1 parent dfb35f6 commit 0ca1d60

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/onnx_opset/generator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,4 +205,4 @@ def version_8(cls, ctx, node, **kwargs):
205205
dtypes = node.output_dtypes
206206
ctx.remove_node(node.name)
207207
for i, output in enumerate(outputs):
208-
ctx.add_graph_input(output, dtypes[i], [-1, -1, -1, -1] if shapes[i] is None else shapes[i])
208+
ctx.add_graph_input(output, dtypes[i], shapes[i])

0 commit comments

Comments
 (0)