onnxruntime
4d39c6a6 - Wire log(softmax) grad cuda kernel and add log(softmax) grad cpu kernel (#4726)

Commit
5 years ago
Wire log(softmax) grad cuda kernel and add log(softmax) grad cpu kernel (#4726) * logsoftmax cuda kernel * add cpu logsoftmaxgrad * revert debug printout * revert disable for debug builds * use /alpha x + y instead * remove misleading log_softmax_ bool Co-authored-by: suffian khan <sukha@OrtTrainingDev1.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
Author
suffiank
Parents
Loading