onnxruntime/onnxruntime
ashbhandare bfbcc89db1
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
2021-05-14 09:00:27 -07:00
..
contrib_ops Add nhwc support for QLinearAveragePool operator (#7656) 2021-05-13 22:05:30 -07:00
core Add MLFloat16 support for SoftmaxCrossEntropyLoss for CUDA EP (#7679) 2021-05-14 09:00:27 -07:00
featurizers_ops/cpu
gsl
python Add ATenOp and call aten::embedding and its Backward Op from ORT (#7590) 2021-05-13 09:24:27 +08:00
test Add MLFloat16 support for SoftmaxCrossEntropyLoss for CUDA EP (#7679) 2021-05-14 09:00:27 -07:00
tool/etw
wasm Bind existing SessionOptions and RunOptions in Javascript API with WebAssembly (#7621) 2021-05-13 10:50:04 -07:00
.style.yapf
__init__.py
ReformatSource.ps1
ReformatSourcePython.bat
VSCodeCoverage.runsettings