Searched refs:AlignGroupsWith (Results 1 – 3 of 3) sorted by relevance
1700 rhs.sharding() == UngroupSharding(AlignGroupsWith( in PartitionDotGroupOnBatch()1712 rhs_grouped = AlignGroupsWith(std::move(rhs_grouped), lhs_grouped); in PartitionDotGroupOnBatch()1715 lhs_grouped = AlignGroupsWith(std::move(lhs_grouped), rhs_grouped); in PartitionDotGroupOnBatch()1720 output_grouped = AlignGroupsWith( in PartitionDotGroupOnBatch()1806 auto grouped = AlignGroupsWith( in PartitionDotGroupOnBatch()1901 return AlignGroupsWith( in GetNonContractingPartitionGroupedShardingForMatchedOperand()1964 return AlignGroupsWith( in GetNonContractingPartitionGroupedShardingForOtherOperand()1972 return AlignGroupsWith( in GetNonContractingPartitionGroupedShardingForOtherOperand()2134 rhs_grouped = AlignGroupsWith(rhs_grouped, lhs_grouped); in PartitionDotGroupOnContracting()2141 lhs_grouped = AlignGroupsWith(lhs_grouped, rhs_grouped); in PartitionDotGroupOnContracting()[all …]
328 GroupedSharding AlignGroupsWith(GroupedSharding grouped_sharding,
1482 GroupedSharding AlignGroupsWith(GroupedSharding grouped_sharding, in AlignGroupsWith() function1553 return UngroupSharding(AlignGroupsWith(sharding_grouped, reference_grouped)); in AlignShardingOnDims()