onnxruntime
Optimize OneHot CUDA Kernel
#4012
Merged

Optimize OneHot CUDA Kernel #4012

Lafi7e merged 4 commits into master from weicwang/onehot
Lafi7e
Lafi7e Optimize for OneHot with zero off value.
3651ea01
Lafi7e Merge branch 'master' into weicwang/onehot
d4e4c662
Lafi7e Add test cases for indices out of range.
7b343567
Lafi7e Lafi7e added training
Lafi7e Lafi7e requested a review from SherlockNoMad SherlockNoMad 5 years ago
Lafi7e Lafi7e requested a review from pengwa pengwa 5 years ago
Lafi7e Lafi7e requested a review from weixingzhang weixingzhang 5 years ago
Lafi7e Lafi7e requested a review 5 years ago
SherlockNoMad
SherlockNoMad approved these changes on 2020-05-25
SherlockNoMad
Merge branch 'master' into weicwang/onehot
90c52f65
Lafi7e Lafi7e merged 9d0534c0 into master 5 years ago
Lafi7e Lafi7e deleted the weicwang/onehot branch 5 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone