Commit 30727a6b authored by Yuanzhong Xu's avatar Yuanzhong Xu Committed by TensorFlower Gardener
Browse files

[XLA] HLO BF16 propagation pass. 

Using BFloat16Support provided by the backend to determine what precision is needed for
each HloInstruction. If the implementation of some HLOs already reduces input precision to BF16, this pass can enable BF16 on more ops without affecting the result.

PiperOrigin-RevId: 186656378
parent 78916e73
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