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
ONNX Export Scripted Interpolate Op #27566
ONNX Export Scripted Interpolate Op #27566
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
@pytorchbot rebase this please |
@houseroad merged this pull request in 735463f. |
Summary: We currently support exporting traced interpolate ops to ONNX. Scripting interpolate op invokes aten::__interpolate in the Torch IR (instead of aten::upsample_[mode][dim]d), which we do not support yet. This PR implements the ONNX symbolic for __interpolate() to support exporting interpolate in scripting scenarios. Related open issue: pytorch#25807 Pull Request resolved: pytorch#27566 Reviewed By: hl475 Differential Revision: D17817731 Pulled By: houseroad fbshipit-source-id: e091793df503e2497f24821cf2954ff157492c75
We currently support exporting traced interpolate ops to ONNX.
Scripting interpolate op invokes aten::_interpolate in the Torch IR (instead of aten::upsample[mode][dim]d), which we do not support yet.
This PR implements the ONNX symbolic for __interpolate() to support exporting interpolate in scripting scenarios.
Related open issue: #25807