int output_rank) {
  Value zp_input_contribution = CreateZeroPointPartialOffsetXlaDotV2(
      builder, loc, input, weight_zp, dnums, /*is_lhs=*/true, output_rank);
  Value zp_weight_contribution = CreateZeroPointPartialOffsetXlaDotV2(
      builder, loc, weight, input_zp, dnums, /*is_lhs=*/false, output_rank);