Commit Graph

3 Commits (a60893f6b550d5dd03002a7b22e35368cd1a5dc3)

Author SHA1 Message Date
wawltor 95aa53425d
update the code for the topk message optimize
5 years ago
LutaoChu 669efb98de
Fix bug: shapes of Topk outputs are wrong when the parameter k is Tensor
5 years ago
wawltor 286eca2d9e
update the code for the topk v2
5 years ago