Searched refs:rhs_with_halo (Results 1 – 1 of 1) sorted by relevance
463 HloInstruction* rhs_with_halo = rhs.hlo(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local486 rhs_with_halo, rhs.base_shape(), left_halo_size_functions[dim], in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()496 rhs_with_halo = *concat; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()500 auto conv, create_sharded_conv(conv_lhs, rhs_with_halo, b, new_window)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()