Skip to content

Commit 055419e

Browse files
authored
Merge pull request #97 from pengwa/fix-placeholder-shape-override
fix shape override issue for placeholder
2 parents 9940b5b + f7082e4 commit 055419e

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)