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