Searched refs:rhs_sharding_dims_adjusted_to_output (Results 1 – 1 of 1) sorted by relevance
1676 auto rhs_sharding_dims_adjusted_to_output = in PartitionDotGroupOnBatch() local1688 rhs_sharding_dims_adjusted_to_output[dim.rhs]) { in PartitionDotGroupOnBatch()1693 rhs_sharding_dims_adjusted_to_output[dim.rhs] = in PartitionDotGroupOnBatch()1858 &rhs_sharding_dims_adjusted_to_output)) { in PartitionDotGroupOnBatch()