Searched refs:source_is_partial_replicate (Results 1 – 1 of 1) sorted by relevance
1229 bool source_is_partial_replicate = sharding().ReplicateOnLastTileDim(); in ReshardPartialReplicateWithAllToAll() local1231 source_is_partial_replicate ? sharding() : target; in ReshardPartialReplicateWithAllToAll()1236 const auto& tile_sharding = source_is_partial_replicate ? target : sharding(); in ReshardPartialReplicateWithAllToAll()1294 if (source_is_partial_replicate) { in ReshardPartialReplicateWithAllToAll()