Searched refs:reshaped_output_tiling (Results 1 – 1 of 1) sorted by relevance
1718 auto reshaped_output_tiling = output_sharding.tile_assignment(); in PartitionDotGroupOnBatch() local1719 reshaped_output_tiling.Reshape(output_sharding_dims_adjusted_to_lhs); in PartitionDotGroupOnBatch()1723 ? HloSharding::PartialTile(reshaped_output_tiling) in PartitionDotGroupOnBatch()1724 : HloSharding::Tile(reshaped_output_tiling), in PartitionDotGroupOnBatch()