Thanks to visit codestin.com
Credit goes to github.com

Skip to content

[ONNX] Support dim=None for argmax and argmin#18264

Closed
zasdfgbnm wants to merge 3 commits into
pytorch:masterfrom
zasdfgbnm:argmaxmin-onnx
Closed

[ONNX] Support dim=None for argmax and argmin#18264
zasdfgbnm wants to merge 3 commits into
pytorch:masterfrom
zasdfgbnm:argmaxmin-onnx

Conversation

@zasdfgbnm
Copy link
Copy Markdown
Collaborator

Fixes: #18263
cc: @houseroad

Copy link
Copy Markdown
Member

@houseroad houseroad left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for adding this support.

Please don't use expect files to test in this case.

Check test/onnx/test_pytorch_onnx_caffe2.py, and do it end to end. :-)

@zasdfgbnm
Copy link
Copy Markdown
Collaborator Author

@houseroad Done

Copy link
Copy Markdown
Contributor

@facebook-github-bot facebook-github-bot left a 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.

@zasdfgbnm zasdfgbnm deleted the argmaxmin-onnx branch March 26, 2019 03:47
@facebook-github-bot
Copy link
Copy Markdown
Contributor

@houseroad merged this pull request in bf2a30c.

laurentdupin pushed a commit to laurentdupin/pytorch that referenced this pull request Apr 24, 2026
Summary:
Fixes: pytorch#18263
cc: houseroad
Pull Request resolved: pytorch#18264

Reviewed By: ezyang

Differential Revision: D14559234

Pulled By: houseroad

fbshipit-source-id: c5b8623752d6c6af41c6d715fd9585a65294868d
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[ONNX] Enable argmax and argmin with no dim and no keepdims

4 participants