xla

2078d797 - [SPMD] index local shards correctly in the execution arguments for replication. (#5110)