Searched defs:lhs_sharding (Results 1 – 3 of 3) sorted by relevance
259 const HloSharding& rhs_sharding) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()572 const HloSharding& rhs_sharding) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
414 const HloSharding& lhs_sharding, const HloSharding& rhs_sharding, in RequiresTransposeSharding()446 const HloSharding& lhs_sharding, const HloSharding& rhs_sharding, in GetWindowedEinsumConfiguration()776 const HloSharding& lhs_sharding = lhs.sharding(); in PartitionBaseCase() local2546 HloSharding lhs_sharding = lhs.sharding(); in GetDotGroupPartitionContractingLhsRhsShardings() local2608 HloSharding lhs_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local3050 HloSharding lhs_sharding = HloSharding::Replicate(); in PrioritizeContractingDimensionsPartitioning() local
430 HloSharding lhs_sharding = HloSharding::Replicate(); in TEST() local