Commit c8e96735 authored by Michael Kuperstein's avatar Michael Kuperstein Committed by TensorFlower Gardener
Browse files

[TF:XLA] A more generic TopK. 

Use Sort to implement R1 TopK for an arbitrary dimension size, and more types.

PiperOrigin-RevId: 202681175
parent e19d4924
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment