Searched refs:left_padded_rhs (Results 1 – 1 of 1) sorted by relevance
553 auto left_padded_rhs = HaloExchangeToPadOnLeft(rhs, reversed_rhs_dims); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local554 if (left_padded_rhs == nullptr) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()557 left_padded_rhs->set_sharding(rhs.sharding()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()558 rhs = PartitionedHlo(left_padded_rhs, rhs.base_shape(), rhs.state()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()