[XLA] Try to pass layouts through reshapes.
For reshapes where the operand and the output have the same rank, try to pass the layout through the reshape. The layout that's already present was presumably assigned for some reason, so it has a good chance of being good. PiperOrigin-RevId: 172555906
Loading
Please sign in to comment