Commit 8b11496c authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Return directly when num_rows is 0 in TopK op. 

When num_rows is 0, SortPairsDescending() called in the GPU kernel will return an error, even though there is no work to do.

PiperOrigin-RevId: 235268833
parent 10247def
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