Commit 6aeb1fdc authored by Benjamin Kramer's avatar Benjamin Kramer Committed by TensorFlower Gardener
Browse files

[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
parent 879fc344
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment