Commit Graph

4 Commits (17030ff28b9a54bb57779e9b8448a6d222110ec5)

Author SHA1 Message Date
jiangcheng 8f4ac6b525
optimize topk op through limit SortTopK kernel entrance, test=develop (#30403)
4 years ago
wawltor 95aa53425d
update the code for the topk message optimize
4 years ago
yongqiangma e4cc6a28b0
Norm op support 2-axis (#26492)
5 years ago
wawltor 286eca2d9e
update the code for the topk v2
5 years ago