Skip to content

Commit f7082e4

Browse files
committed
fix shape override issue for placeholder
1 parent 9940b5b commit f7082e4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/tfonnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,7 @@ def reduce_op(ctx, node, name, args):
235235

236236

237237
def placeholder_op(ctx, node, name, args):
238-
input_node = helper.make_tensor_value_info(node.output[0], node.dtype, node.shape)
238+
input_node = helper.make_tensor_value_info(node.output[0], node.dtype, ctx.get_shape(node.output[0]))
239239
ctx.model_inputs.append(input_node)
240240
return None
241241

0 commit comments

Comments
 (0)