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

[Disscussion] Would mechanism like onnx "register_schema" be helpful in optimizing models with user-defined operators? #326

Open
SStarver opened this issue Apr 3, 2024 · 0 comments

Comments

@SStarver
Copy link

SStarver commented Apr 3, 2024

Recently, I have discovered that the ONNX project has introduced the "onnx.defs.register_schema(op_schema)" interface, enabling support for shape inference in models that utilize user-defined operators. This addition is not only beneficial but crucial in preventing errors that arise during the model verification process, especially when dealing with customized operators.

Given this, I am intrigued to explore whether this functionality could aid in the optimization of models that contain customized operations. Currently, attempts to optimize models with such operators often result in failures, often due to unrecognized operators.

  • case without register_schema
    With "onnx.defs.register_schema(op_schema)" unnoted, following errors would be gone. Futhermore, shape_inference would also gone futher.
    image
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