-
Notifications
You must be signed in to change notification settings - Fork 3.8k
Closed
Labels
Description
Bug Report
Describe the bug
import onnx
import onnx.parser
model = """
<
ir_version: 9,
opset_import: ["" : 11]
>
graph (float[7,6,1,5] in0, float in1, float[7,2,3,2,1] in2) => () {
out0 = Conv <auto_pad = "NOTSET", group = 1> (in0, in1, in2)
}
"""
onnx.shape_inference.infer_shapes(onnx.parser.parse_model(model))
crashes with a segmentation fault.
System information
- OS Platform and Distribution (. Linux Ubuntu 20.04): Linux Ubuntu 20.04
- ONNX version (e.g. 1.13): 1.16.1
- Python version: 3.10.12
Expected behavior
No crash, but an error that the model is invalid