Searched defs:rhs_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()530 const HloSharding& rhs_sharding = rhs.sharding(); in PartitionBaseCase() local2230 HloSharding rhs_sharding = rhs.sharding(); in GetDotGroupPartitionContractingLhsRhsShardings() local2291 HloSharding rhs_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local2675 HloSharding rhs_sharding = HloSharding::Replicate(); in PrioritizeContractingDimensionsPartitioning() local