Searched refs:unsupported_sharding (Results 1 – 1 of 1) sorted by relevance
258 auto unsupported_sharding = [&](const HloSharding& lhs_sharding, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local271 if (unsupported_sharding(aligned_lhs_sharding, rhs.sharding())) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()277 if (unsupported_sharding(lhs.sharding(), aligned_rhs_sharding)) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()569 auto unsupported_sharding = [&](const HloSharding& lhs_sharding, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local580 if (unsupported_sharding(aligned_lhs_sharding, rhs.sharding())) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()586 if (unsupported_sharding(lhs.sharding(), aligned_rhs_sharding)) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()