pytorch
2458b3cd - [MPS] Add argmin (#80828)

Commit
2 years ago
[MPS] Add argmin (#80828) This PR 1. adds argmin 2. refactors `reduction_type` in `ReduceOps.mm` with enum. Co-authored by Kulin Seth <kulinseth@gmail.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/80828 Approved by: https://github.com/malfet
Author
Committer
Parents
Loading