Go
pytorch
4615f6aa
- [MPS]: Add fix for squeezed input axes handling in BCE loss (#79676)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
2 years ago
[MPS]: Add fix for squeezed input axes handling in BCE loss (#79676) Fixes #79527 Pull Request resolved: https://github.com/pytorch/pytorch/pull/79676 Approved by: https://github.com/razarmehr, https://github.com/albanD
Author
kulinseth
Committer
pytorchmergebot
Parents
e79a51f7
Loading