search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.49k stars 480 forks source link

Fixed masked_scatter and masked_select #8037

Closed anishfish2 closed 2 months ago

anishfish2 commented 2 months ago

Implementing masked_scatter and masked_select op. This PR should mark conclusion of all ops in #7510.