Searched refs:to_adjust (Results 1 – 1 of 1) sorted by relevance
1874 HloSharding* to_adjust = &new_operand_shard; in GatherOperandsShardedAcrossParallelDims() local1877 std::swap(to_adjust, target); in GatherOperandsShardedAcrossParallelDims()1879 if (!to_adjust->ReplicateOnLastTileDim()) { in GatherOperandsShardedAcrossParallelDims()1882 auto new_tile_assignment_dims = to_adjust->tile_assignment().dimensions(); in GatherOperandsShardedAcrossParallelDims()1885 int64 to_adjust_dim = to_adjust->tile_assignment().dim(to_adjust_dims[i]); in GatherOperandsShardedAcrossParallelDims()1906 auto new_tile_assignment = to_adjust->tile_assignment(); in GatherOperandsShardedAcrossParallelDims()1909 *to_adjust = in GatherOperandsShardedAcrossParallelDims()1913 *to_adjust = AlignShardingOnDims(HloSharding::Tile(new_tile_assignment), in GatherOperandsShardedAcrossParallelDims()