onnxruntime/orttraining
Vincent Wang f0f3012666
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
2021-06-09 10:29:46 +08:00
..
orttraining Add SoftmaxCrossEntropyLossInternal to Support Dynamic ignore_index Input (#7899) 2021-06-09 10:29:46 +08:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools add dockfile for ROCm 4.2 (#7749) 2021-06-08 08:02:27 -07:00