Searched refs:conv_lhs (Results 1 – 1 of 1) sorted by relevance
453 HloInstruction* conv_lhs = lhs.hlo(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local457 conv_lhs = b->AddInstruction(HloInstruction::CreateDynamicSlice( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()500 auto conv, create_sharded_conv(conv_lhs, rhs_with_halo, b, new_window)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()