xla

780a8231 - Transfer shards from server using replication (#4508)

Transfer shards from server using replication (#4508) * [SPMD] Transfer shards back to host with replication * [SPMD] Add test_transfer_sharded_data_to_host test case.