Skip to content

Commit

Permalink
Fix wrong comments
Browse files Browse the repository at this point in the history
Signed-off-by: Yuhong Guo <yuhong.gyh@antgroup.com>
  • Loading branch information
Yuhong Guo committed Jun 30, 2023
1 parent d879a5e commit 6ceda4d
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion onnx/test/shape_inference_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -446,7 +446,7 @@ def test_cast(self, _, version) -> None:
graph,
[make_tensor_value_info("y", TensorProto.UINT8, (2, 4, 3))],
opset_imports=[helper.make_opsetid(ONNX_DOMAIN, version)],
# Concat Version 1 does not have shape inference function.
# Cast Version 1 does not have shape inference function.
has_inference_function=has_inference_function("Cast", version),
)

Expand Down Expand Up @@ -615,6 +615,7 @@ def test_concat(self, _, version) -> None:
graph,
[make_tensor_value_info("z", TensorProto.FLOAT, (9, 4, 3))],
opset_imports=[helper.make_opsetid(ONNX_DOMAIN, version)],
# Concat Version 1 does not have shape inference function.
has_inference_function=has_inference_function("Concat", version),
)

Expand Down Expand Up @@ -718,6 +719,7 @@ def test_reshape_dynamic_shape_symbolic(self, _, version) -> None:
graph,
[make_tensor_value_info("y", TensorProto.UINT8, None)],
opset_imports=[helper.make_opsetid(ONNX_DOMAIN, version)],
# Reshape Version 1 does not have shape inference function.
has_inference_function=has_inference_function("Reshape", version),
)

Expand Down Expand Up @@ -881,6 +883,7 @@ def test_upsample(self, _, version) -> None:
graph,
[make_tensor_value_info("y", TensorProto.INT32, (2, 4, 3, 9))],
opset_imports=[helper.make_opsetid(ONNX_DOMAIN, version)],
# Upsample Version 1 does not have shape inference function.
has_inference_function=has_inference_function("Upsample", version),
)
else:
Expand Down

0 comments on commit 6ceda4d

Please sign in to comment.