Commit 45a841ae authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Handle reduce-window in bfloat16 propagation 

If the reduce computation's is precision preserving (e.g. max) then the
reduce window is precision preserving as well so we can propagate
bfloat16 through it.

PiperOrigin-RevId: 220783083
parent 5c07cebb
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