Searched refs:per_group_lhs (Results 1 – 1 of 1) sorted by relevance
1705 PartitionedHlo per_group_lhs = lhs; in PartitionDotGroupOnBatch() local1735 per_group_lhs = PartitionedHlo( in PartitionDotGroupOnBatch()1850 per_group_lhs = *resharded; in PartitionDotGroupOnBatch()1864 per_group_lhs.sharding() == per_group_rhs.sharding()); in PartitionDotGroupOnBatch()1868 PartitionDot(per_group_lhs, per_group_rhs, in PartitionDotGroupOnBatch()