You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
import onnx
import onnx.shape_inference
import onnx.inliner
model = onnx.load("maf_mnist.onnx")
onnx.checker.check_model(model)
print("Pass checker!")
# Works fine
onnx.shape_inference.infer_shapes(model, data_prop=False)
print("Success with data_prop=False !")
inlined_model = onnx.inliner.inline_local_functions(model)
print("Success with inlining!")
onnx.shape_inference.infer_shapes(inlined_model, data_prop=True)
print("Success with data_prop=True on inlined model !")
# Segfaults
onnx.shape_inference.infer_shapes(model, data_prop=True)