Commit 1a306c94 authored by Yuanzhong Xu's avatar Yuanzhong Xu Committed by TensorFlower Gardener
Browse files

[XLA] Fix BF16 propagation type adjustment 

AllUsersConsumeBF16() incorrectly used ValueTypeAfterChange() for the current value being checked, but it should be the original type.

Also fusion computation should be adjusted as soon as the fusion root is adjusted.

There was also redundant work for while computations. Now removed.

PiperOrigin-RevId: 206216822
parent 6f4e330e
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