onnxruntime
GatherElementsGrad CPU Kernel and TopKGrad CPU/CUDA Kernel
#5511
Merged

GatherElementsGrad CPU Kernel and TopKGrad CPU/CUDA Kernel #5511

Lafi7e merged 3 commits into master from weicwang/topkgradcpu
Lafi7e
Lafi7e TopKGrad CPU kernel
20748b96
Lafi7e Lafi7e added training
Lafi7e Lafi7e requested a review from mrry mrry 5 years ago
Lafi7e Lafi7e requested a review from SherlockNoMad SherlockNoMad 5 years ago
Lafi7e Lafi7e requested a review from nbcsm nbcsm 5 years ago
Lafi7e Lafi7e requested a review 5 years ago
SherlockNoMad
SherlockNoMad commented on 2020-10-16
SherlockNoMad
SherlockNoMad commented on 2020-10-16
SherlockNoMad
SherlockNoMad commented on 2020-10-16
Lafi7e use Scatter for GatherElementsGrad and TopKGrad.
c2392746
Lafi7e Lafi7e changed the title TopKGrad CPU kernel GatherElementsGrad CPU Kernel and TopKGrad CPU/CUDA Kernel 5 years ago
Lafi7e rollback convgrad change.
29c2df0c
mrry
mrry approved these changes on 2020-10-20
Lafi7e Lafi7e merged b48f596a into master 5 years ago
Lafi7e Lafi7e deleted the weicwang/topkgradcpu branch 5 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone