[V1][TPU] Enable Top K #15489
NickLucche
marked this pull request as ready for review 1 year ago
NickLucche
marked this pull request as draft 1 year ago
NickLucche
marked this pull request as ready for review 1 year ago
enable topk
7baa0e80
[V1][TPU] TPU-optimized top-k implementation (2nd try)
0a81a65b
random k in test + fix rebase
71700b58
simplify test
339650c8
topk equivalence tests
ee0cd585
revert long
e62ca74d
mgoin
approved these changes
on 2025-04-17
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub