-
Notifications
You must be signed in to change notification settings - Fork 3.7k
add arg test plus a few minor fixes #1049
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
@@ -18,7 +18,7 @@ def export_nearest(): # type: () -> None | |||
'Upsample', | |||
inputs=['x'], | |||
outputs=['y'], | |||
scales=[1.0, 1.0, 2.0, 3.0], | |||
scales=[2.0, 3.0], | |||
mode='nearest', |
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.
no. it was discussed and agreed to keep the element number of scales the same as the rank of inputs. In this test case, the input's rank is 4, and we don't want to upsample the 1st and 2nd dimension, so it's 1.0. Make sense?
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.
As the upsample op now support multiple-D input tensor now (instead of 4-D tensor previously).
fix type annotation, undo upsample change add arg test plus a few minor fixes fix type annotation, undo upsample change Polish the test case
) Summary: Pass the end to end test cases in onnx/onnx#1049 Closes #9050 Reviewed By: hlu1 Differential Revision: D8703757 Pulled By: houseroad fbshipit-source-id: 63308202e349dfc02d532e87f49495ba1aab085b
…torch#9050) Summary: Pass the end to end test cases in onnx/onnx#1049 Closes pytorch#9050 Reviewed By: hlu1 Differential Revision: D8703757 Pulled By: houseroad fbshipit-source-id: 63308202e349dfc02d532e87f49495ba1aab085b
* add arg test plus a few minor fixes fix type annotation, undo upsample change add arg test plus a few minor fixes fix type annotation, undo upsample change Polish the test case * Update test cases for ArgMax and ArgMin
* add arg test plus a few minor fixes fix type annotation, undo upsample change add arg test plus a few minor fixes fix type annotation, undo upsample change Polish the test case * Update test cases for ArgMax and ArgMin
No description provided.