[XLA] Fix BF16 propagation pass to produce matching fusion root and output.
Previously, the propagation pass might produce different procision in the fused computation's root than the fusion itself, when the fused root doesn't define a buffer. Add explicit converts at such fusion roots. PiperOrigin-RevId: 186812368
Loading
Please sign in to comment