onnxruntime
e945b5fc
- adding fp16 support for topk cuda kernel (#6082)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
adding fp16 support for topk cuda kernel (#6082) * adding fp16 support for topk. * disable fp16 tests for cpu ep Co-authored-by: Du Li <duli@OrtTrainingDev0.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
References
#6082 - adding fp16 support for topk cuda kernel
Author
duli2012
Parents
7ddeafdf
Loading