Commit 6455940a authored by Benjamin Kramer's avatar Benjamin Kramer Committed by TensorFlower Gardener
Browse files

[XLA:GPU] Consolidate shape checking for multi-output loop fusion again 

This is supposed to be checked by ShapesCompatibleForFusion, but it didn't take
non-input fusion reduces into account.

PiperOrigin-RevId: 210072191
parent a5eefeb6
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