onnxruntime
Fix softmax_warp_backward math when is_log_softmax = True and register LogSoftmax CUDA kernel
#5160

Merged

Fix softmax_warp_backward math when is_log_softmax = True and register LogSoftmax CUDA kernel #5160

suffiank merged 3 commits into master from sukha/registerlogsoftmax

5fc26777

forgot to remove axis check

8891e6e7

suffiank requested a review 5 years ago

SherlockNoMad commented on 2020-09-14

add tests all axis

c22f0a7c

suffiank added training

suffiank added release:orttraining_rc3

SherlockNoMad approved these changes on 2020-09-17

suffiank closed this 5 years ago

suffiank reopened this 5 years ago

suffiank merged e01e0b2e into master 5 years ago

suffiank deleted the sukha/registerlogsoftmax branch 5 years ago

edgchen1 added triage:approved

suffiank added release:1.5.0

faxu removed release:1.5.0

Reviewers

SherlockNoMad

Assignees

No one assigned

Labels

training

Milestone

No milestone