onnxruntime
bfbcc89d - Add MLFloat16 support for SoftmaxCrossEntropyLoss for CUDA EP (#7679)

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
Author
Parents
Loading