pytorch
6a14fcb9 - [MPS] Add support for aten::masked_select on mps (#119) (#85818)

Commit
3 years ago
[MPS] Add support for aten::masked_select on mps (#119) (#85818) Reuse the `index.Tensor_out` implementation since it's already expanding the bool/byte indices to long tensors. Pull Request resolved: https://github.com/pytorch/pytorch/pull/85818 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading