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()
393 const HloSharding& lhs_sharding, const HloSharding& rhs_sharding) { in GetWindowedEinsumConfiguration()484 const HloSharding& lhs_sharding = lhs.sharding(); in PartitionBaseCase() local2098 auto lhs_sharding = lhs.sharding(); in PartitionDotGroupOnContracting() local