[XLA:GPU] Fuse broadcasts into reduction fusions
We didn't do this because reconstructing a layout was hard. With layout_assignment before fusion this becomes much easier. Remove the limitations. PiperOrigin-RevId: 188167436
Loading
Please sign in to comment