xla
ca7297d3 - Add one_hot to ATen XLA tensor

Commit
6 years ago
Add one_hot to ATen XLA tensor It's sufficient to just route it to at::native::one_hot. Still goes back to host for range validation of labels, which cannot be avoided.
Author
Committer
Parents
Loading