Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[BUG] Simplify removes local functions from the ONNX model #320

Open
aernoudt opened this issue Feb 5, 2024 · 1 comment
Open

[BUG] Simplify removes local functions from the ONNX model #320

aernoudt opened this issue Feb 5, 2024 · 1 comment

Comments

@aernoudt
Copy link

aernoudt commented Feb 5, 2024

Describe the bug
onnxsim.simplify (version 0.4.35) removes local functions from the model

Model
The issue can be reproduced with following script: testLocalFunction.py.txt

After running onnxsim.simplify the local function MyReshapeFunction is removed from the model and inference fails.
Commenting out lines 69 and 70 makes the inference work.

@aernoudt
Copy link
Author

@daquexian any idea what causes this issue?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant