Add support for ArgMax and ArgMin in C2 onnx backend and frontend#9050
Add support for ArgMax and ArgMin in C2 onnx backend and frontend#9050houseroad wants to merge 3 commits intopytorch:masterfrom
Conversation
facebook-github-bot
left a comment
There was a problem hiding this comment.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
facebook-github-bot
left a comment
There was a problem hiding this comment.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@pytorchbot retest this please |
1 similar comment
|
@pytorchbot retest this please |
caffe2/onnx/onnx_exporter.cc
Outdated
| auto& node = nodes.back(); | ||
|
|
||
| bool explicit_axis = false; | ||
| for (const auto& a : def.arg()) { |
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.
facebook-github-bot
left a comment
There was a problem hiding this comment.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
facebook-github-bot
left a comment
There was a problem hiding this comment.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@pytorchbot retest this please |
…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
Pass the end to end test cases in onnx/onnx#1049