onnxruntime

e01e0b2e - Fix softmax_warp_backward math when is_log_softmax = True and register LogSoftmax CUDA kernel (#5160)

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>