Searched refs:tgt_tile_assignment (Results 1 – 1 of 1) sorted by relevance
1820 auto tgt_tile_assignment = source_sharding.tile_assignment(); in CreateMatchingShardingOnDims() local1821 tgt_tile_assignment.Reshape(tile_dims); in CreateMatchingShardingOnDims()1823 return AlignShardingOnDims(HloSharding::PartialTile(tgt_tile_assignment), in CreateMatchingShardingOnDims()1826 return AlignShardingOnDims(HloSharding::Tile(tgt_tile_assignment), in CreateMatchingShardingOnDims()