onnxruntime
70e2f96e
- Support double for operator TopK + fix one bug in TopK implementation for GPU for double (#6220)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
Support double for operator TopK + fix one bug in TopK implementation for GPU for double (#6220) * Support double for operator TopK * add static classes for topk/double * fix cast issue in topk
References
#6220 - Support double for operator TopK + fix one bug in TopK implementation for GPU for double
Author
xadupre
Parents
ecb2e119
Loading