onnxruntime
70e2f96e - Support double for operator TopK + fix one bug in TopK implementation for GPU for double (#6220)

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
Author
Parents
Loading