[TF:XLA] Separate on-host and on-device shape and layout in HloModule.
Previously, only one layout was stored with an HLO module. This CL allows HLO passes to modify the on-device layouts without affecting the on-host layout (provided by the client) PiperOrigin-RevId: 195014875
Loading
Please sign in to comment