Add ONNX symbolic for UpsamplingNearest2d#3450
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.
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.