Searched refs:lhs_with_halo (Results 1 – 1 of 1) sorted by relevance
657 HloInstruction* lhs_with_halo = lhs.hlo(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local713 lhs_with_halo, lhs.base_shape(), left_halo_size_functions[dim], in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()723 lhs_with_halo = *concat; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()727 auto conv, create_sharded_conv(lhs_with_halo, rhs.hlo(), b, new_window)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()