Skip to content

Commit 7bc260f

Browse files
committed
add comment for v5
1 parent adb3fff commit 7bc260f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/onnx_opset/tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1478,7 +1478,7 @@ def version_10(cls, ctx, node, **kwargs):
14781478
slice_op = GraphBuilder(ctx).make_slice({"data": new_nonmaxsurppress,
14791479
"axes": [1], "ends": [3], "starts": [2]})
14801480
squeeze_op = ctx.make_node("Squeeze", [slice_op], attr={"axes": [1]})
1481-
if len(node.input) > 5: # V5
1481+
if len(node.input) > 5: # v5, called by ..._with_scores(), pad_to_max_output_size always False
14821482
ctx.make_node("Cast", inputs=squeeze_op.output, attr={"to": onnx_pb.TensorProto.INT32},
14831483
outputs=[node.output[0]], dtypes=dtypes[0], shapes=shapes[0])
14841484
ctx.make_node("Gather", inputs=[input_score.input[0], squeeze_op.output[0]],

0 commit comments

Comments
 (0)