Searched refs:lhs_sharding_dims_adjusted_to_output (Results 1 – 1 of 1) sorted by relevance
1672 auto lhs_sharding_dims_adjusted_to_output = in PartitionDotGroupOnBatch() local1687 if (lhs_sharding_dims_adjusted_to_output[dim.lhs] != in PartitionDotGroupOnBatch()1691 lhs_sharding_dims_adjusted_to_output[dim.lhs] = in PartitionDotGroupOnBatch()1849 &lhs_sharding_dims_adjusted_to_output)) { in PartitionDotGroupOnBatch()