onnxruntime
f0f30126
- Add SoftmaxCrossEntropyLossInternal to Support Dynamic ignore_index Input (#7899)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
4 years ago
Add SoftmaxCrossEntropyLossInternal to Support Dynamic ignore_index Input (#7899) * add SoftmaxCrossEntropyLossInternal * bugfix and ut * fix ut * fix ut * support torch1.8.1 * function body for nll_loss_internal
References
#7899 - Add SoftmaxCrossEntropyLossInternal to Support Dynamic ignore_index Input
Author
Lafi7e
Parents
f38200e2
Loading