onnxruntime
bfbcc89d
- Add MLFloat16 support for SoftmaxCrossEntropyLoss for CUDA EP (#7679)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
4 years ago
Add MLFloat16 support for SoftmaxCrossEntropyLoss for CUDA EP (#7679) * Forward op changes * Add tests, improve kernel * add opset 13 registration, remove unnecessary changes * Add fp16 grad for SCELoss, review comments
References
#7679 - Add MLFloat16 support for SoftmaxCrossEntropyLoss for CUDA EP
Author
ashbhandare
Parents
39fac6d3
Loading