-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Add ONNX symbolic for UpsamplingNearest2d #3450
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
Conversation
c874a9a to
2a517c0
Compare
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
ezyang
left a comment
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.
Use the helper function here. Also, we should export this as a Caffe2 vendor extension op while it is not supported in ONNX (talk to me about how to do this.)
|
Just realize that this one is a little bit tricky, since ATen has no corresponding implementation. |
2a517c0 to
3779101
Compare
|
onnx/onnx#180, now we have it as an experimental op. Later we may polish it. I think upsample op is useful/necessary. |
|
Build finished. |
1 similar comment
|
Build finished. |
|
I think Dima already updated the symbolic. But it got changed in ONNX, so I will update this PR to fix the problem. |
3779101 to
62ba644
Compare
62ba644 to
3d15ef3
Compare
Add the symbolic for UpsamplingNearest2d.
Locally tested on test_operators.py and test_caffe2.py.
Will update ONNX and ONNX-Caffe2 later.