onnxruntime/orttraining
Suffian Khan 39a7f96a44 Fix softmax_warp_backward math when is_log_softmax = True and register LogSoftmax CUDA kernel (#5160)
* register logsoftmax cuda kernel; fix logsoftmaxgrad cuda kernal; fix tests to invoke dispatch_softmax_*

* forgot to remove axis check

* add tests all axis

Co-authored-by: suffian khan <sukha@OrtTrainingDev1.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
2020-09-23 16:50:58 -07:00
..
orttraining Fix softmax_warp_backward math when is_log_softmax = True and register LogSoftmax CUDA kernel (#5160) 2020-09-23 16:50:58 -07:00
pytorch_frontend_examples Fix mnist example (#4926) 2020-08-26 15:28:39 -07:00
tools Update convergence baseline for ci_test. (#4465) 2020-07-09 15:29:36 +08:00