search

pytorch / xla

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

Implement bucketize in torch aten ops (#7396) #8007

Closed simonteozw closed 2 months ago