Searched refs:operand_shard_tiles (Results 1 – 1 of 1) sorted by relevance
1925 auto operand_shard_tiles = new_operand_shard.tile_assignment(); in GatherOperandsShardedAcrossParallelDims() local1926 operand_shard_tiles.Reshape(operand_shard_tile_dims); in GatherOperandsShardedAcrossParallelDims()1929 ? HloSharding::PartialTile(operand_shard_tiles) in GatherOperandsShardedAcrossParallelDims()1930 : HloSharding::Tile(operand_shard_tiles), in GatherOperandsShardedAcrossParallelDims()