auto nmsv4 = CreateTfOp<TF::NonMaxSuppressionV4Op>(
        rewriter, nms_op, new_result_types, nms_op.getBoxes(),
        nms_op.getScores(), nms_op.getMaxOutputSize(), nms_op.getIouThreshold(),
        nms_op.getScoreThreshold());
    // Cannot replace the NMSv3 Op with NMSv4 since the outputs between the
    // two are different (v4 expects two output values vs v3 requires only one.