def METHOD_NAME(self, model: ModelProto, input_name: str) -> ValueInfoProto:
    """
    Get the input proto for the specified input name.
    :param model: the ONNX model
    :param input_name: the name of the input
    """
    input_proto = [
        input_info
        for input_info in model.graph.input
        if input_info.name == input_name
    ][0]
    if not input_proto:
        raise RuntimeError(
            f"{self.__name__} - unable to find '{input_name}' in model input"
        )
    return input_proto