xla
ed0cdd84
- Added aten::upsample_nearest2d and aten::upsample_nearest2d_backward for TPU.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
6 years ago
Added aten::upsample_nearest2d and aten::upsample_nearest2d_backward for TPU. Going to PyTorch CPU for XLA CPU/GPU, as those have not implemented the XLA custom-call to lower them.
References
#1233 - Added aten::upsample_nearest2d and aten::upsample_nearest2d_backward …
Author
dlibenzi
Committer
dlibenzi
Parents
79ff8634
Loading