Searched refs:output_sharding_dims_adjusted_to_lhs (Results 1 – 1 of 1) sorted by relevance
1680 auto output_sharding_dims_adjusted_to_lhs = in PartitionDotGroupOnBatch() local1695 output_sharding_dims_adjusted_to_lhs[dim.output] = in PartitionDotGroupOnBatch()1719 reshaped_output_tiling.Reshape(output_sharding_dims_adjusted_to_lhs); in PartitionDotGroupOnBatch()