Shape Inference with Custom Operators

[erickksun]

hi,
I have an ONNX model that includes a custom operator, which I implemented using ONNX Runtime Extension. The implementation functions as expected. However, I am facing an issue with performing shape inference and saving the results back into the ONNX file. I’ve attempted to use the tool found at this link, but it appears not to support custom operators.

[skottmckay]

Why do you need to save the results back to the ONNX file? Shape inferencing runs during model load, so if the shapes can be inferred there's generally no need to save them previously.