Searched refs:tmp_tile_assignment (Results 1 – 1 of 1) sorted by relevance
1285 auto tmp_tile_assignment = tile_sharding.tile_assignment(); in ReshardPartialReplicateWithAllToAll() local1290 tmp_tile_assignment.Reshape(tmp_tile_assignment_dimensions); in ReshardPartialReplicateWithAllToAll()1292 HloSharding::PartialTile(tmp_tile_assignment); in ReshardPartialReplicateWithAllToAll()