Go
pytorch
37ebcc8a
- [MPS] Add argmin (#80828)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
2 years ago
[MPS] Add argmin (#80828) This PR 1. adds argmin 2. refactors `reduction_type` in `ReduceOps.mm` with enum. Pull Request resolved: https://github.com/pytorch/pytorch/pull/80828 Approved by: https://github.com/malfet
Author
qqaatw
Committer
pytorchmergebot
Parents
7f37b1b3
Loading