[XLA:GPU] Allow intermediate outputs for reduce input fusions.
This generalizes the emitter to allow pretty much arbitrary multi-output fusion as long as the shapes match the input of the reduce(s). The idea is that multi-output fusion can move intermediate inputs into the same fusion so they don't have to be re-read by the reduce. PiperOrigin-RevId: 199497832
Loading
Please sign in to comment