Commit 2e21bcf1 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Avoid constant folding partial output of a node. It fixes issues #22383. Its... 

Avoid constant folding partial output of a node. It fixes issues #22383. Its root cause is that tf only fold values but not indices when running TopK on GPU with dtype float. TopK generates different results on GPU and CPU.

PiperOrigin-RevId: 221657401
parent c7d57d43
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