xla
ed0cdd84 - Added aten::upsample_nearest2d and aten::upsample_nearest2d_backward for TPU.

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.
Author
Committer
Parents
Loading