Searched refs:per_group_rhs (Results 1 – 1 of 1) sorted by relevance
1706 PartitionedHlo per_group_rhs = rhs; in PartitionDotGroupOnBatch() local1738 per_group_rhs = PartitionedHlo( in PartitionDotGroupOnBatch()1859 per_group_rhs = *resharded; in PartitionDotGroupOnBatch()1864 per_group_lhs.sharding() == per_group_rhs.sharding()); in PartitionDotGroupOnBatch()1868 PartitionDot(per_group_lhs, per_group_rhs, in PartitionDotGroupOnBatch()