topi.rst
4.93 KB
-
[Relay/TOPI][Op] Add TopK operator (#3256) · 072f8cc7
* init impl for topk * Fix cpu for topk * init cuda impl for topk * Add cuda for topk * fix * Add doc * update doc * lint * lint * lint * x * fix warning * [Relay] Add TopK in tf converter * Add frontend converter * fix
Haichen Shen committed