onnxruntime
adding fp16 support for topk cuda kernel
#6082
Merged

adding fp16 support for topk cuda kernel #6082

duli2012 merged 4 commits into master from duli/topk
duli2012
adding fp16 support for topk.
73675da4
duli2012 duli2012 requested a review from hariharans29 hariharans29 5 years ago
duli2012 duli2012 requested a review from HectorSVC HectorSVC 5 years ago
duli2012 duli2012 requested a review 5 years ago
hanbitmyths
hanbitmyths commented on 2020-12-08
hariharans29
hariharans29 commented on 2020-12-08
yuslepukhin
yuslepukhin commented on 2020-12-09
adding fp16 support for topk.
b371a91a
Merge branch 'duli/topk' of https://github.com/microsoft/onnxruntime …
79f73dd1
hanbitmyths
disable fp16 tests for cpu ep
1b281979
duli2012
hanbitmyths
hanbitmyths approved these changes on 2020-12-10
duli2012 duli2012 merged e945b5fc into master 5 years ago
duli2012 duli2012 deleted the duli/topk branch 5 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone