Searched defs:lhs_sharding (Results 1 – 2 of 2) sorted by relevance
259 const HloSharding& rhs_sharding) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()570 const HloSharding& rhs_sharding) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
387 const HloSharding& lhs_sharding, const HloSharding& rhs_sharding) { in GetWindowedEinsumConfiguration()529 const HloSharding& lhs_sharding = lhs.sharding(); in PartitionBaseCase() local2229 HloSharding lhs_sharding = lhs.sharding(); in GetDotGroupPartitionContractingLhsRhsShardings() local2290 HloSharding lhs_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local2674 HloSharding lhs_sharding = HloSharding::Replicate(); in PrioritizeContractingDimensionsPartitioning() local