xla
b6e6d863
- Add cosine_embedding_loss to ATen XLA type
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
6 years ago
Add cosine_embedding_loss to ATen XLA type It's sufficient to just route it to at::native::cosine_embedding_loss.
References
#565 - Add most of the remaining loss, norm and distance functions
Author
asuhan
Committer
asuhan
Parents
baf1a766
Loading