Go
pytorch
ae83e44c
- [MPS] Handle 1D inputs for NLL (#81290)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
2 years ago
[MPS] Handle 1D inputs for NLL (#81290) * Add test for NLL 1d * Fix forward NLL for 1D case * Handle NLL backward for 1d Pull Request resolved: https://github.com/pytorch/pytorch/pull/81290 Approved by: https://github.com/razarmehr
Author
abhudev
Committer
pytorchmergebot
Parents
478081c6
Loading