Commit 7586bb3c authored by Adrian Kuegel's avatar Adrian Kuegel Committed by TensorFlower Gardener
Browse files

Simplify TopK xla client op. 

Instead of creating a rank 1 iota and broadcasting it, we can directly create
the iota we need.

PiperOrigin-RevId: 222377479
parent f1b77734
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