onnx 1.15.0__cp311-cp311-win_amd64.whl → 1.16.1__cp311-cp311-win_amd64.whl

onnx/onnx_cpp2py_export/defs.pyi

@@ -0,0 +1,202 @@
+"""Submodule containing all the ONNX schema definitions."""
+from __future__ import annotations
+
+from typing import Sequence, overload
+
+from onnx import AttributeProto, FunctionProto
+
+class SchemaError(Exception): ...
+
+class OpSchema:
+    def __init__(
+        self,
+        name: str,
+        domain: str,
+        since_version: int,
+        doc: str = "",
+        *,
+        inputs: Sequence[OpSchema.FormalParameter] = (),
+        outputs: Sequence[OpSchema.FormalParameter] = (),
+        type_constraints: Sequence[tuple[str, Sequence[str], str]] = (),
+        attributes: Sequence[OpSchema.Attribute] = (),
+    ) -> None: ...
+    @property
+    def file(self) -> str: ...
+    @property
+    def line(self) -> int: ...
+    @property
+    def support_level(self) -> SupportType: ...
+    @property
+    def doc(self) -> str | None: ...
+    @property
+    def since_version(self) -> int: ...
+    @property
+    def deprecated(self) -> bool: ...
+    @property
+    def domain(self) -> str: ...
+    @property
+    def name(self) -> str: ...
+    @property
+    def min_input(self) -> int: ...
+    @property
+    def max_input(self) -> int: ...
+    @property
+    def min_output(self) -> int: ...
+    @property
+    def max_output(self) -> int: ...
+    @property
+    def attributes(self) -> dict[str, Attribute]: ...
+    @property
+    def inputs(self) -> Sequence[FormalParameter]: ...
+    @property
+    def outputs(self) -> Sequence[FormalParameter]: ...
+    @property
+    def type_constraints(self) -> Sequence[TypeConstraintParam]: ...
+    @property
+    def has_type_and_shape_inference_function(self) -> bool: ...
+    @property
+    def has_data_propagation_function(self) -> bool: ...
+    @staticmethod
+    def is_infinite(v: int) -> bool: ...
+    def consumed(self, schema: OpSchema, i: int) -> tuple[UseType, int]: ...
+    def _infer_node_outputs(
+        self,
+        node_proto: bytes,
+        value_types: dict[str, bytes],
+        input_data: dict[str, bytes],
+        input_sparse_data: dict[str, bytes],
+    ) -> dict[str, bytes]: ...
+    @property
+    def function_body(self) -> FunctionProto: ...
+
+    class TypeConstraintParam:
+        def __init__(
+            self,
+            type_param_str: str,
+            allowed_type_strs: Sequence[str],
+            description: str = "",
+        ) -> None:
+            """Type constraint parameter.
+
+            Args:
+                type_param_str: Type parameter string, for example, "T", "T1", etc.
+                allowed_type_strs: Allowed type strings for this type parameter. E.g. ["tensor(float)"].
+                description: Type parameter description.
+            """
+        @property
+        def type_param_str(self) -> str: ...
+        @property
+        def description(self) -> str: ...
+        @property
+        def allowed_type_strs(self) -> Sequence[str]: ...
+
+    class FormalParameterOption:
+        Single: OpSchema.FormalParameterOption = ...
+        Optional: OpSchema.FormalParameterOption = ...
+        Variadic: OpSchema.FormalParameterOption = ...
+
+    class DifferentiationCategory:
+        Unknown: OpSchema.DifferentiationCategory = ...
+        Differentiable: OpSchema.DifferentiationCategory = ...
+        NonDifferentiable: OpSchema.DifferentiationCategory = ...
+
+    class FormalParameter:
+        def __init__(
+            self,
+            name: str,
+            type_str: str,
+            description: str = "",
+            *,
+            param_option: OpSchema.FormalParameterOption = OpSchema.FormalParameterOption.Single,  # noqa: F821
+            is_homogeneous: bool = True,
+            min_arity: int = 1,
+            differentiation_category: OpSchema.DifferentiationCategory = OpSchema.DifferentiationCategory.Unknown,  # noqa: F821
+        ) -> None: ...
+        @property
+        def name(self) -> str: ...
+        @property
+        def types(self) -> set[str]: ...
+        @property
+        def type_str(self) -> str: ...
+        @property
+        def description(self) -> str: ...
+        @property
+        def option(self) -> OpSchema.FormalParameterOption: ...
+        @property
+        def is_homogeneous(self) -> bool: ...
+        @property
+        def min_arity(self) -> int: ...
+        @property
+        def differentiation_category(self) -> OpSchema.DifferentiationCategory: ...
+
+    class AttrType:
+        FLOAT: OpSchema.AttrType = ...
+        INT: OpSchema.AttrType = ...
+        STRING: OpSchema.AttrType = ...
+        TENSOR: OpSchema.AttrType = ...
+        GRAPH: OpSchema.AttrType = ...
+        SPARSE_TENSOR: OpSchema.AttrType = ...
+        TYPE_PROTO: OpSchema.AttrType = ...
+        FLOATS: OpSchema.AttrType = ...
+        INTS: OpSchema.AttrType = ...
+        STRINGS: OpSchema.AttrType = ...
+        TENSORS: OpSchema.AttrType = ...
+        GRAPHS: OpSchema.AttrType = ...
+        SPARSE_TENSORS: OpSchema.AttrType = ...
+        TYPE_PROTOS: OpSchema.AttrType = ...
+
+    class Attribute:
+        @overload
+        def __init__(
+            self,
+            name: str,
+            type: OpSchema.AttrType,  # noqa: A002
+            description: str = "",
+            *,
+            required: bool = True,
+        ) -> None: ...
+        @overload
+        def __init__(
+            self,
+            name: str,
+            default_value: AttributeProto,
+            description: str = "",
+        ) -> None: ...
+        @property
+        def name(self) -> str: ...
+        @property
+        def description(self) -> str: ...
+        @property
+        def type(self) -> OpSchema.AttrType: ...
+        @property
+        def default_value(self) -> AttributeProto: ...
+        @property
+        def required(self) -> bool: ...
+
+    class SupportType(int):
+        COMMON: OpSchema.SupportType = ...
+        EXPERIMENTAL: OpSchema.SupportType = ...
+
+    class UseType:
+        DEFAULT: OpSchema.UseType = ...
+        CONSUME_ALLOWED: OpSchema.UseType = ...
+        CONSUME_ENFORCED: OpSchema.UseType = ...
+
+@overload
+def has_schema(op_type: str, domain: str = "") -> bool: ...
+@overload
+def has_schema(
+    op_type: str, max_inclusive_version: int, domain: str = ""
+) -> bool: ...
+def schema_version_map() -> dict[str, tuple[int, int]]: ...
+@overload
+def get_schema(
+    op_type: str, max_inclusive_version: int, domain: str = ""
+) -> OpSchema: ...
+@overload
+def get_schema(op_type: str, domain: str = "") -> OpSchema: ...
+def get_all_schemas() -> Sequence[OpSchema]: ...
+def get_all_schemas_with_history() -> Sequence[OpSchema]: ...
+def set_domain_to_version(domain: str, min_version: int, max_version: int, last_release_version: int = -1) -> None: ...
+def register_schema(schema: OpSchema) -> None: ...
+def deregister_schema(op_type: str, version: int, domain: str) -> None: ...

onnx/onnx_cpp2py_export/inliner.pyi

@@ -0,0 +1,19 @@
+# Copyright (c) ONNX Project Contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+def inline_local_functions(model: bytes, convert_version: bool) -> bytes:
+    """Inlines calls to model-local function in input model and returns it.
+    Both input and output are serialized ModelProtos.
+    """
+    ...
+
+def inline_selected_functions(model: bytes, function_ids: list[tuple[str,str]], exclude: bool) -> bytes:
+    """Inlines calls to selected model-local functions in input model and returns it.
+    Inlines all functions specified in function_ids, unless exclude is true, in which
+    case it inlines all functions except those specified in function_ids.
+    Both input and output are serialized ModelProtos.
+    """
+    ...

onnx/onnx_cpp2py_export/parser.pyi

@@ -0,0 +1,32 @@
+
+def parse_model(model: str) -> tuple[bool, bytes, bytes]:
+    """Returns (success-flag, error-message, serialized-proto).
+
+    If success-flag is true, then serialized-proto contains the parsed ModelProto.
+    Otherwise, error-message contains a string describing the parse error.
+    """
+    ...
+
+def parse_graph(graph: str) -> tuple[bool, bytes, bytes]:
+    """Returns (success-flag, error-message, serialized-proto).
+
+    If success-flag is true, then serialized-proto contains the parsed GraphProto.
+    Otherwise, error-message contains a string describing the parse error.
+    """
+    ...
+
+def parse_function(function: str) -> tuple[bool, bytes, bytes]:
+    """Returns (success-flag, error-message, serialized-proto).
+
+    If success-flag is true, then serialized-proto contains the parsed FunctionProto.
+    Otherwise, error-message contains a string describing the parse error.
+    """
+    ...
+
+def parse_node(node: str) -> tuple[bool, bytes, bytes]:
+    """Returns (success-flag, error-message, serialized-proto).
+
+    If success-flag is true, then serialized-proto contains the parsed NodeProto.
+    Otherwise, error-message contains a string describing the parse error.
+    """
+    ...

onnx/onnx_cpp2py_export/printer.pyi

@@ -0,0 +1,3 @@
+def function_to_text(serialized_function_proto: bytes) -> str: ...
+def graph_to_text(serialized_graph_proto: bytes) -> str: ...
+def model_to_text(serialized_model_proto: bytes) -> str: ...

onnx/onnx_cpp2py_export/shape_inference.pyi

@@ -0,0 +1,16 @@
+
+class InferenceError(Exception): ...
+
+def infer_shapes(
+    b: bytes, check_type: bool, strict_mode: bool, data_prop: bool
+) -> bytes: ...
+
+def infer_shapes_path(
+    model_path: str,
+    output_path: str,
+    check_type: bool,
+    strict_mode: bool,
+    data_prop: bool,
+) -> None: ...
+
+def infer_function_output_types(bytes: bytes, input_types: list[bytes], attributes: list[bytes]) -> list[bytes]: ...  # noqa: A002

onnx/onnx_cpp2py_export/version_converter.pyi

@@ -0,0 +1,4 @@
+class ConvertError(Exception): ...
+
+# Where the first bytes are a serialized ModelProto
+def convert_version(bytes: bytes, target: int) -> bytes: ...  # noqa: A002

onnx/onnx_data_pb2.pyi

@@ -0,0 +1,146 @@
+# @generated by protoc-gen-mypy.py.  Do not edit!
+# mypy: disable-error-code=override
+from google.protobuf.message import (  # type: ignore
+    Message,
+)
+
+from typing import (
+    Iterable,
+    List,
+    Optional as OptionalType,
+    Tuple,
+    cast,
+)
+
+from google.protobuf.internal.containers import (  # type: ignore
+    RepeatedCompositeFieldContainer,
+    RepeatedScalarFieldContainer,
+)
+
+from onnx.onnx_ml_pb2 import (
+    SparseTensorProto,
+    TensorProto,
+)
+
+class SequenceProto(Message):
+    class DataType(int):
+        @classmethod
+        def Name(cls, number: int) -> str: ...
+        @classmethod
+        def Value(cls, name: str) -> int: ...
+        @classmethod
+        def keys(cls) -> List[str]: ...
+        @classmethod
+        def values(cls) -> List[int]: ...
+        @classmethod
+        def items(cls) -> List[Tuple[str, int]]: ...
+    UNDEFINED = cast(DataType, 0)
+    TENSOR = cast(DataType, 1)
+    SPARSE_TENSOR = cast(DataType, 2)
+    SEQUENCE = cast(DataType, 3)
+    MAP = cast(DataType, 4)
+    OPTIONAL = cast(DataType, 5)
+    
+    name = ... # type: str
+    elem_type = ... # type: int
+    
+    @property
+    def tensor_values(self) -> RepeatedCompositeFieldContainer[TensorProto]: ...
+    
+    @property
+    def sparse_tensor_values(self) -> RepeatedCompositeFieldContainer[SparseTensorProto]: ...
+    
+    @property
+    def sequence_values(self) -> RepeatedCompositeFieldContainer[SequenceProto]: ...
+    
+    @property
+    def map_values(self) -> RepeatedCompositeFieldContainer[MapProto]: ...
+    
+    @property
+    def optional_values(self) -> RepeatedCompositeFieldContainer[OptionalProto]: ...
+    
+    def __init__(self,
+        name : OptionalType[str] = None,
+        elem_type : OptionalType[int] = None,
+        tensor_values : OptionalType[Iterable[TensorProto]] = None,
+        sparse_tensor_values : OptionalType[Iterable[SparseTensorProto]] = None,
+        sequence_values : OptionalType[Iterable[SequenceProto]] = None,
+        map_values : OptionalType[Iterable[MapProto]] = None,
+        optional_values : OptionalType[Iterable[OptionalProto]] = None,
+        ) -> None: ...
+    @classmethod
+    def FromString(cls, s: bytes) -> SequenceProto: ...
+    def MergeFrom(self, other_msg: Message) -> None: ...
+    def CopyFrom(self, other_msg: Message) -> None: ...
+
+class MapProto(Message):
+    name = ... # type: str
+    key_type = ... # type: int
+    keys = ... # type: RepeatedScalarFieldContainer[int]
+    string_keys = ... # type: RepeatedScalarFieldContainer[bytes]
+    
+    @property
+    def values(self) -> SequenceProto: ...
+    
+    def __init__(self,
+        name : OptionalType[str] = None,
+        key_type : OptionalType[int] = None,
+        keys : OptionalType[Iterable[int]] = None,
+        string_keys : OptionalType[Iterable[bytes]] = None,
+        values : OptionalType[SequenceProto] = None,
+        ) -> None: ...
+    @classmethod
+    def FromString(cls, s: bytes) -> MapProto: ...
+    def MergeFrom(self, other_msg: Message) -> None: ...
+    def CopyFrom(self, other_msg: Message) -> None: ...
+
+class OptionalProto(Message):
+    class DataType(int):
+        @classmethod
+        def Name(cls, number: int) -> str: ...
+        @classmethod
+        def Value(cls, name: str) -> int: ...
+        @classmethod
+        def keys(cls) -> List[str]: ...
+        @classmethod
+        def values(cls) -> List[int]: ...
+        @classmethod
+        def items(cls) -> List[Tuple[str, int]]: ...
+    UNDEFINED = cast(DataType, 0)
+    TENSOR = cast(DataType, 1)
+    SPARSE_TENSOR = cast(DataType, 2)
+    SEQUENCE = cast(DataType, 3)
+    MAP = cast(DataType, 4)
+    OPTIONAL = cast(DataType, 5)
+    
+    name = ... # type: str
+    elem_type = ... # type: int
+    
+    @property
+    def tensor_value(self) -> TensorProto: ...
+    
+    @property
+    def sparse_tensor_value(self) -> SparseTensorProto: ...
+    
+    @property
+    def sequence_value(self) -> SequenceProto: ...
+    
+    @property
+    def map_value(self) -> MapProto: ...
+    
+    @property
+    def optional_value(self) -> OptionalProto: ...
+    
+    def __init__(self,
+        name : OptionalType[str] = None,
+        elem_type : OptionalType[int] = None,
+        tensor_value : OptionalType[TensorProto] = None,
+        sparse_tensor_value : OptionalType[SparseTensorProto] = None,
+        sequence_value : OptionalType[SequenceProto] = None,
+        map_value : OptionalType[MapProto] = None,
+        optional_value : OptionalType[OptionalProto] = None,
+        ) -> None: ...
+    @classmethod
+    def FromString(cls, s: bytes) -> OptionalProto: ...
+    def MergeFrom(self, other_msg: Message) -> None: ...
+    def CopyFrom(self, other_msg: Message) -> None: ...

onnx/onnx_ml_pb2.py

@@ -13,7 +13,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12onnx/onnx-ml.proto\x12\x04onnx\"\xcc\x05\n\x0e\x41ttributeProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x15\n\rref_attr_name\x18\x15 \x01(\t\x12\x12\n\ndoc_string\x18\r \x01(\t\x12\x30\n\x04type\x18\x14 \x01(\x0e\x32\".onnx.AttributeProto.AttributeType\x12\t\n\x01\x66\x18\x02 \x01(\x02\x12\t\n\x01i\x18\x03 \x01(\x03\x12\t\n\x01s\x18\x04 \x01(\x0c\x12\x1c\n\x01t\x18\x05 \x01(\x0b\x32\x11.onnx.TensorProto\x12\x1b\n\x01g\x18\x06 \x01(\x0b\x32\x10.onnx.GraphProto\x12.\n\rsparse_tensor\x18\x16 \x01(\x0b\x32\x17.onnx.SparseTensorProto\x12\x1b\n\x02tp\x18\x0e \x01(\x0b\x32\x0f.onnx.TypeProto\x12\x0e\n\x06\x66loats\x18\x07 \x03(\x02\x12\x0c\n\x04ints\x18\x08 \x03(\x03\x12\x0f\n\x07strings\x18\t \x03(\x0c\x12\"\n\x07tensors\x18\n \x03(\x0b\x32\x11.onnx.TensorProto\x12 \n\x06graphs\x18\x0b \x03(\x0b\x32\x10.onnx.GraphProto\x12/\n\x0esparse_tensors\x18\x17 \x03(\x0b\x32\x17.onnx.SparseTensorProto\x12$\n\x0btype_protos\x18\x0f \x03(\x0b\x32\x0f.onnx.TypeProto\"\xd9\x01\n\rAttributeType\x12\r\n\tUNDEFINED\x10\x00\x12\t\n\x05\x46LOAT\x10\x01\x12\x07\n\x03INT\x10\x02\x12\n\n\x06STRING\x10\x03\x12\n\n\x06TENSOR\x10\x04\x12\t\n\x05GRAPH\x10\x05\x12\x11\n\rSPARSE_TENSOR\x10\x0b\x12\x0e\n\nTYPE_PROTO\x10\r\x12\n\n\x06\x46LOATS\x10\x06\x12\x08\n\x04INTS\x10\x07\x12\x0b\n\x07STRINGS\x10\x08\x12\x0b\n\x07TENSORS\x10\t\x12\n\n\x06GRAPHS\x10\n\x12\x12\n\x0eSPARSE_TENSORS\x10\x0c\x12\x0f\n\x0bTYPE_PROTOS\x10\x0e\"Q\n\x0eValueInfoProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x1d\n\x04type\x18\x02 \x01(\x0b\x32\x0f.onnx.TypeProto\x12\x12\n\ndoc_string\x18\x03 \x01(\t\"\x96\x01\n\tNodeProto\x12\r\n\x05input\x18\x01 \x03(\t\x12\x0e\n\x06output\x18\x02 \x03(\t\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x0f\n\x07op_type\x18\x04 \x01(\t\x12\x0e\n\x06\x64omain\x18\x07 \x01(\t\x12\'\n\tattribute\x18\x05 \x03(\x0b\x32\x14.onnx.AttributeProto\x12\x12\n\ndoc_string\x18\x06 \x01(\t\"\xd6\x01\n\x11TrainingInfoProto\x12(\n\x0einitialization\x18\x01 \x01(\x0b\x32\x10.onnx.GraphProto\x12#\n\talgorithm\x18\x02 \x01(\x0b\x32\x10.onnx.GraphProto\x12<\n\x16initialization_binding\x18\x03 \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\x12\x34\n\x0eupdate_binding\x18\x04 \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\"\xeb\x02\n\nModelProto\x12\x12\n\nir_version\x18\x01 \x01(\x03\x12.\n\x0copset_import\x18\x08 \x03(\x0b\x32\x18.onnx.OperatorSetIdProto\x12\x15\n\rproducer_name\x18\x02 \x01(\t\x12\x18\n\x10producer_version\x18\x03 \x01(\t\x12\x0e\n\x06\x64omain\x18\x04 \x01(\t\x12\x15\n\rmodel_version\x18\x05 \x01(\x03\x12\x12\n\ndoc_string\x18\x06 \x01(\t\x12\x1f\n\x05graph\x18\x07 \x01(\x0b\x32\x10.onnx.GraphProto\x12\x34\n\x0emetadata_props\x18\x0e \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\x12.\n\rtraining_info\x18\x14 \x03(\x0b\x32\x17.onnx.TrainingInfoProto\x12&\n\tfunctions\x18\x19 \x03(\x0b\x32\x13.onnx.FunctionProto\"4\n\x16StringStringEntryProto\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"k\n\x10TensorAnnotation\x12\x13\n\x0btensor_name\x18\x01 \x01(\t\x12\x42\n\x1cquant_parameter_tensor_names\x18\x02 \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\"\x9e\x03\n\nGraphProto\x12\x1d\n\x04node\x18\x01 \x03(\x0b\x32\x0f.onnx.NodeProto\x12\x0c\n\x04name\x18\x02 \x01(\t\x12&\n\x0binitializer\x18\x05 \x03(\x0b\x32\x11.onnx.TensorProto\x12\x33\n\x12sparse_initializer\x18\x0f \x03(\x0b\x32\x17.onnx.SparseTensorProto\x12\x12\n\ndoc_string\x18\n \x01(\t\x12#\n\x05input\x18\x0b \x03(\x0b\x32\x14.onnx.ValueInfoProto\x12$\n\x06output\x18\x0c \x03(\x0b\x32\x14.onnx.ValueInfoProto\x12(\n\nvalue_info\x18\r \x03(\x0b\x32\x14.onnx.ValueInfoProto\x12\x37\n\x17quantization_annotation\x18\x0e \x03(\x0b\x32\x16.onnx.TensorAnnotationJ\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x06\x10\nR\nir_versionR\x10producer_versionR\x0cproducer_tagR\x06\x64omain\"\x82\x06\n\x0bTensorProto\x12\x0c\n\x04\x64ims\x18\x01 \x03(\x03\x12\x11\n\tdata_type\x18\x02 \x01(\x05\x12*\n\x07segment\x18\x03 \x01(\x0b\x32\x19.onnx.TensorProto.Segment\x12\x16\n\nfloat_data\x18\x04 \x03(\x02\x42\x02\x10\x01\x12\x16\n\nint32_data\x18\x05 \x03(\x05\x42\x02\x10\x01\x12\x13\n\x0bstring_data\x18\x06 \x03(\x0c\x12\x16\n\nint64_data\x18\x07 \x03(\x03\x42\x02\x10\x01\x12\x0c\n\x04name\x18\x08 \x01(\t\x12\x12\n\ndoc_string\x18\x0c \x01(\t\x12\x10\n\x08raw_data\x18\t \x01(\x0c\x12\x33\n\rexternal_data\x18\r \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\x12\x35\n\rdata_location\x18\x0e \x01(\x0e\x32\x1e.onnx.TensorProto.DataLocation\x12\x17\n\x0b\x64ouble_data\x18\n \x03(\x01\x42\x02\x10\x01\x12\x17\n\x0buint64_data\x18\x0b \x03(\x04\x42\x02\x10\x01\x1a%\n\x07Segment\x12\r\n\x05\x62\x65gin\x18\x01 \x01(\x03\x12\x0b\n\x03\x65nd\x18\x02 \x01(\x03\"\xa4\x02\n\x08\x44\x61taType\x12\r\n\tUNDEFINED\x10\x00\x12\t\n\x05\x46LOAT\x10\x01\x12\t\n\x05UINT8\x10\x02\x12\x08\n\x04INT8\x10\x03\x12\n\n\x06UINT16\x10\x04\x12\t\n\x05INT16\x10\x05\x12\t\n\x05INT32\x10\x06\x12\t\n\x05INT64\x10\x07\x12\n\n\x06STRING\x10\x08\x12\x08\n\x04\x42OOL\x10\t\x12\x0b\n\x07\x46LOAT16\x10\n\x12\n\n\x06\x44OUBLE\x10\x0b\x12\n\n\x06UINT32\x10\x0c\x12\n\n\x06UINT64\x10\r\x12\r\n\tCOMPLEX64\x10\x0e\x12\x0e\n\nCOMPLEX128\x10\x0f\x12\x0c\n\x08\x42\x46LOAT16\x10\x10\x12\x10\n\x0c\x46LOAT8E4M3FN\x10\x11\x12\x12\n\x0e\x46LOAT8E4M3FNUZ\x10\x12\x12\x0e\n\nFLOAT8E5M2\x10\x13\x12\x12\n\x0e\x46LOAT8E5M2FNUZ\x10\x14\")\n\x0c\x44\x61taLocation\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x0c\n\x08\x45XTERNAL\x10\x01\"h\n\x11SparseTensorProto\x12!\n\x06values\x18\x01 \x01(\x0b\x32\x11.onnx.TensorProto\x12\"\n\x07indices\x18\x02 \x01(\x0b\x32\x11.onnx.TensorProto\x12\x0c\n\x04\x64ims\x18\x03 \x03(\x03\"\x95\x01\n\x10TensorShapeProto\x12-\n\x03\x64im\x18\x01 \x03(\x0b\x32 .onnx.TensorShapeProto.Dimension\x1aR\n\tDimension\x12\x13\n\tdim_value\x18\x01 \x01(\x03H\x00\x12\x13\n\tdim_param\x18\x02 \x01(\tH\x00\x12\x12\n\ndenotation\x18\x03 \x01(\tB\x07\n\x05value\"\xa5\x05\n\tTypeProto\x12-\n\x0btensor_type\x18\x01 \x01(\x0b\x32\x16.onnx.TypeProto.TensorH\x00\x12\x31\n\rsequence_type\x18\x04 \x01(\x0b\x32\x18.onnx.TypeProto.SequenceH\x00\x12\'\n\x08map_type\x18\x05 \x01(\x0b\x32\x13.onnx.TypeProto.MapH\x00\x12\x31\n\roptional_type\x18\t \x01(\x0b\x32\x18.onnx.TypeProto.OptionalH\x00\x12:\n\x12sparse_tensor_type\x18\x08 \x01(\x0b\x32\x1c.onnx.TypeProto.SparseTensorH\x00\x12-\n\x0bopaque_type\x18\x07 \x01(\x0b\x32\x16.onnx.TypeProto.OpaqueH\x00\x12\x12\n\ndenotation\x18\x06 \x01(\t\x1a\x42\n\x06Tensor\x12\x11\n\telem_type\x18\x01 \x01(\x05\x12%\n\x05shape\x18\x02 \x01(\x0b\x32\x16.onnx.TensorShapeProto\x1a.\n\x08Sequence\x12\"\n\telem_type\x18\x01 \x01(\x0b\x32\x0f.onnx.TypeProto\x1a<\n\x03Map\x12\x10\n\x08key_type\x18\x01 \x01(\x05\x12#\n\nvalue_type\x18\x02 \x01(\x0b\x32\x0f.onnx.TypeProto\x1a.\n\x08Optional\x12\"\n\telem_type\x18\x01 \x01(\x0b\x32\x0f.onnx.TypeProto\x1aH\n\x0cSparseTensor\x12\x11\n\telem_type\x18\x01 \x01(\x05\x12%\n\x05shape\x18\x02 \x01(\x0b\x32\x16.onnx.TensorShapeProto\x1a&\n\x06Opaque\x12\x0e\n\x06\x64omain\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\tB\x07\n\x05value\"5\n\x12OperatorSetIdProto\x12\x0e\n\x06\x64omain\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\x03\"\x94\x02\n\rFunctionProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05input\x18\x04 \x03(\t\x12\x0e\n\x06output\x18\x05 \x03(\t\x12\x11\n\tattribute\x18\x06 \x03(\t\x12-\n\x0f\x61ttribute_proto\x18\x0b \x03(\x0b\x32\x14.onnx.AttributeProto\x12\x1d\n\x04node\x18\x07 \x03(\x0b\x32\x0f.onnx.NodeProto\x12\x12\n\ndoc_string\x18\x08 \x01(\t\x12.\n\x0copset_import\x18\t \x03(\x0b\x32\x18.onnx.OperatorSetIdProto\x12\x0e\n\x06\x64omain\x18\n \x01(\tJ\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04R\rsince_versionR\x06status*\xfe\x01\n\x07Version\x12\x12\n\x0e_START_VERSION\x10\x00\x12\x19\n\x15IR_VERSION_2017_10_10\x10\x01\x12\x19\n\x15IR_VERSION_2017_10_30\x10\x02\x12\x18\n\x14IR_VERSION_2017_11_3\x10\x03\x12\x18\n\x14IR_VERSION_2019_1_22\x10\x04\x12\x18\n\x14IR_VERSION_2019_3_18\x10\x05\x12\x18\n\x14IR_VERSION_2019_9_19\x10\x06\x12\x17\n\x13IR_VERSION_2020_5_8\x10\x07\x12\x18\n\x14IR_VERSION_2021_7_30\x10\x08\x12\x0e\n\nIR_VERSION\x10\t*.\n\x0eOperatorStatus\x12\x10\n\x0c\x45XPERIMENTAL\x10\x00\x12\n\n\x06STABLE\x10\x01\x42\x02H\x03')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12onnx/onnx-ml.proto\x12\x04onnx\"\xdb\x05\n\x0e\x41ttributeProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x15\n\rref_attr_name\x18\x15 \x01(\t\x12\x12\n\ndoc_string\x18\r \x01(\t\x12\x30\n\x04type\x18\x14 \x01(\x0e\x32\".onnx.AttributeProto.AttributeType\x12\t\n\x01\x66\x18\x02 \x01(\x02\x12\t\n\x01i\x18\x03 \x01(\x03\x12\t\n\x01s\x18\x04 \x01(\x0c\x12\x1c\n\x01t\x18\x05 \x01(\x0b\x32\x11.onnx.TensorProto\x12\x1b\n\x01g\x18\x06 \x01(\x0b\x32\x10.onnx.GraphProto\x12.\n\rsparse_tensor\x18\x16 \x01(\x0b\x32\x17.onnx.SparseTensorProto\x12\x1b\n\x02tp\x18\x0e \x01(\x0b\x32\x0f.onnx.TypeProto\x12\x0e\n\x06\x66loats\x18\x07 \x03(\x02\x12\x0c\n\x04ints\x18\x08 \x03(\x03\x12\x0f\n\x07strings\x18\t \x03(\x0c\x12\"\n\x07tensors\x18\n \x03(\x0b\x32\x11.onnx.TensorProto\x12 \n\x06graphs\x18\x0b \x03(\x0b\x32\x10.onnx.GraphProto\x12/\n\x0esparse_tensors\x18\x17 \x03(\x0b\x32\x17.onnx.SparseTensorProto\x12$\n\x0btype_protos\x18\x0f \x03(\x0b\x32\x0f.onnx.TypeProto\"\xd9\x01\n\rAttributeType\x12\r\n\tUNDEFINED\x10\x00\x12\t\n\x05\x46LOAT\x10\x01\x12\x07\n\x03INT\x10\x02\x12\n\n\x06STRING\x10\x03\x12\n\n\x06TENSOR\x10\x04\x12\t\n\x05GRAPH\x10\x05\x12\x11\n\rSPARSE_TENSOR\x10\x0b\x12\x0e\n\nTYPE_PROTO\x10\r\x12\n\n\x06\x46LOATS\x10\x06\x12\x08\n\x04INTS\x10\x07\x12\x0b\n\x07STRINGS\x10\x08\x12\x0b\n\x07TENSORS\x10\t\x12\n\n\x06GRAPHS\x10\n\x12\x12\n\x0eSPARSE_TENSORS\x10\x0c\x12\x0f\n\x0bTYPE_PROTOS\x10\x0eJ\x04\x08\x0c\x10\rJ\x04\x08\x10\x10\x14R\x01v\"\x87\x01\n\x0eValueInfoProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x1d\n\x04type\x18\x02 \x01(\x0b\x32\x0f.onnx.TypeProto\x12\x12\n\ndoc_string\x18\x03 \x01(\t\x12\x34\n\x0emetadata_props\x18\x04 \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\"\xde\x01\n\tNodeProto\x12\r\n\x05input\x18\x01 \x03(\t\x12\x0e\n\x06output\x18\x02 \x03(\t\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x0f\n\x07op_type\x18\x04 \x01(\t\x12\x0e\n\x06\x64omain\x18\x07 \x01(\t\x12\x10\n\x08overload\x18\x08 \x01(\t\x12\'\n\tattribute\x18\x05 \x03(\x0b\x32\x14.onnx.AttributeProto\x12\x12\n\ndoc_string\x18\x06 \x01(\t\x12\x34\n\x0emetadata_props\x18\t \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\"\xd6\x01\n\x11TrainingInfoProto\x12(\n\x0einitialization\x18\x01 \x01(\x0b\x32\x10.onnx.GraphProto\x12#\n\talgorithm\x18\x02 \x01(\x0b\x32\x10.onnx.GraphProto\x12<\n\x16initialization_binding\x18\x03 \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\x12\x34\n\x0eupdate_binding\x18\x04 \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\"\xeb\x02\n\nModelProto\x12\x12\n\nir_version\x18\x01 \x01(\x03\x12.\n\x0copset_import\x18\x08 \x03(\x0b\x32\x18.onnx.OperatorSetIdProto\x12\x15\n\rproducer_name\x18\x02 \x01(\t\x12\x18\n\x10producer_version\x18\x03 \x01(\t\x12\x0e\n\x06\x64omain\x18\x04 \x01(\t\x12\x15\n\rmodel_version\x18\x05 \x01(\x03\x12\x12\n\ndoc_string\x18\x06 \x01(\t\x12\x1f\n\x05graph\x18\x07 \x01(\x0b\x32\x10.onnx.GraphProto\x12\x34\n\x0emetadata_props\x18\x0e \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\x12.\n\rtraining_info\x18\x14 \x03(\x0b\x32\x17.onnx.TrainingInfoProto\x12&\n\tfunctions\x18\x19 \x03(\x0b\x32\x13.onnx.FunctionProto\"4\n\x16StringStringEntryProto\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"k\n\x10TensorAnnotation\x12\x13\n\x0btensor_name\x18\x01 \x01(\t\x12\x42\n\x1cquant_parameter_tensor_names\x18\x02 \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\"\xd4\x03\n\nGraphProto\x12\x1d\n\x04node\x18\x01 \x03(\x0b\x32\x0f.onnx.NodeProto\x12\x0c\n\x04name\x18\x02 \x01(\t\x12&\n\x0binitializer\x18\x05 \x03(\x0b\x32\x11.onnx.TensorProto\x12\x33\n\x12sparse_initializer\x18\x0f \x03(\x0b\x32\x17.onnx.SparseTensorProto\x12\x12\n\ndoc_string\x18\n \x01(\t\x12#\n\x05input\x18\x0b \x03(\x0b\x32\x14.onnx.ValueInfoProto\x12$\n\x06output\x18\x0c \x03(\x0b\x32\x14.onnx.ValueInfoProto\x12(\n\nvalue_info\x18\r \x03(\x0b\x32\x14.onnx.ValueInfoProto\x12\x37\n\x17quantization_annotation\x18\x0e \x03(\x0b\x32\x16.onnx.TensorAnnotation\x12\x34\n\x0emetadata_props\x18\x10 \x03(\x0b\x32\x1c.onnx.StringStringEntryProtoJ\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x06\x10\nR\nir_versionR\x10producer_versionR\x0cproducer_tagR\x06\x64omain\"\xcd\x06\n\x0bTensorProto\x12\x0c\n\x04\x64ims\x18\x01 \x03(\x03\x12\x11\n\tdata_type\x18\x02 \x01(\x05\x12*\n\x07segment\x18\x03 \x01(\x0b\x32\x19.onnx.TensorProto.Segment\x12\x16\n\nfloat_data\x18\x04 \x03(\x02\x42\x02\x10\x01\x12\x16\n\nint32_data\x18\x05 \x03(\x05\x42\x02\x10\x01\x12\x13\n\x0bstring_data\x18\x06 \x03(\x0c\x12\x16\n\nint64_data\x18\x07 \x03(\x03\x42\x02\x10\x01\x12\x0c\n\x04name\x18\x08 \x01(\t\x12\x12\n\ndoc_string\x18\x0c \x01(\t\x12\x10\n\x08raw_data\x18\t \x01(\x0c\x12\x33\n\rexternal_data\x18\r \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\x12\x35\n\rdata_location\x18\x0e \x01(\x0e\x32\x1e.onnx.TensorProto.DataLocation\x12\x17\n\x0b\x64ouble_data\x18\n \x03(\x01\x42\x02\x10\x01\x12\x17\n\x0buint64_data\x18\x0b \x03(\x04\x42\x02\x10\x01\x12\x34\n\x0emetadata_props\x18\x10 \x03(\x0b\x32\x1c.onnx.StringStringEntryProto\x1a%\n\x07Segment\x12\r\n\x05\x62\x65gin\x18\x01 \x01(\x03\x12\x0b\n\x03\x65nd\x18\x02 \x01(\x03\"\xb9\x02\n\x08\x44\x61taType\x12\r\n\tUNDEFINED\x10\x00\x12\t\n\x05\x46LOAT\x10\x01\x12\t\n\x05UINT8\x10\x02\x12\x08\n\x04INT8\x10\x03\x12\n\n\x06UINT16\x10\x04\x12\t\n\x05INT16\x10\x05\x12\t\n\x05INT32\x10\x06\x12\t\n\x05INT64\x10\x07\x12\n\n\x06STRING\x10\x08\x12\x08\n\x04\x42OOL\x10\t\x12\x0b\n\x07\x46LOAT16\x10\n\x12\n\n\x06\x44OUBLE\x10\x0b\x12\n\n\x06UINT32\x10\x0c\x12\n\n\x06UINT64\x10\r\x12\r\n\tCOMPLEX64\x10\x0e\x12\x0e\n\nCOMPLEX128\x10\x0f\x12\x0c\n\x08\x42\x46LOAT16\x10\x10\x12\x10\n\x0c\x46LOAT8E4M3FN\x10\x11\x12\x12\n\x0e\x46LOAT8E4M3FNUZ\x10\x12\x12\x0e\n\nFLOAT8E5M2\x10\x13\x12\x12\n\x0e\x46LOAT8E5M2FNUZ\x10\x14\x12\t\n\x05UINT4\x10\x15\x12\x08\n\x04INT4\x10\x16\")\n\x0c\x44\x61taLocation\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x0c\n\x08\x45XTERNAL\x10\x01\"h\n\x11SparseTensorProto\x12!\n\x06values\x18\x01 \x01(\x0b\x32\x11.onnx.TensorProto\x12\"\n\x07indices\x18\x02 \x01(\x0b\x32\x11.onnx.TensorProto\x12\x0c\n\x04\x64ims\x18\x03 \x03(\x03\"\x95\x01\n\x10TensorShapeProto\x12-\n\x03\x64im\x18\x01 \x03(\x0b\x32 .onnx.TensorShapeProto.Dimension\x1aR\n\tDimension\x12\x13\n\tdim_value\x18\x01 \x01(\x03H\x00\x12\x13\n\tdim_param\x18\x02 \x01(\tH\x00\x12\x12\n\ndenotation\x18\x03 \x01(\tB\x07\n\x05value\"\xa5\x05\n\tTypeProto\x12-\n\x0btensor_type\x18\x01 \x01(\x0b\x32\x16.onnx.TypeProto.TensorH\x00\x12\x31\n\rsequence_type\x18\x04 \x01(\x0b\x32\x18.onnx.TypeProto.SequenceH\x00\x12\'\n\x08map_type\x18\x05 \x01(\x0b\x32\x13.onnx.TypeProto.MapH\x00\x12\x31\n\roptional_type\x18\t \x01(\x0b\x32\x18.onnx.TypeProto.OptionalH\x00\x12:\n\x12sparse_tensor_type\x18\x08 \x01(\x0b\x32\x1c.onnx.TypeProto.SparseTensorH\x00\x12-\n\x0bopaque_type\x18\x07 \x01(\x0b\x32\x16.onnx.TypeProto.OpaqueH\x00\x12\x12\n\ndenotation\x18\x06 \x01(\t\x1a\x42\n\x06Tensor\x12\x11\n\telem_type\x18\x01 \x01(\x05\x12%\n\x05shape\x18\x02 \x01(\x0b\x32\x16.onnx.TensorShapeProto\x1a.\n\x08Sequence\x12\"\n\telem_type\x18\x01 \x01(\x0b\x32\x0f.onnx.TypeProto\x1a<\n\x03Map\x12\x10\n\x08key_type\x18\x01 \x01(\x05\x12#\n\nvalue_type\x18\x02 \x01(\x0b\x32\x0f.onnx.TypeProto\x1a.\n\x08Optional\x12\"\n\telem_type\x18\x01 \x01(\x0b\x32\x0f.onnx.TypeProto\x1aH\n\x0cSparseTensor\x12\x11\n\telem_type\x18\x01 \x01(\x05\x12%\n\x05shape\x18\x02 \x01(\x0b\x32\x16.onnx.TensorShapeProto\x1a&\n\x06Opaque\x12\x0e\n\x06\x64omain\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\tB\x07\n\x05value\"5\n\x12OperatorSetIdProto\x12\x0e\n\x06\x64omain\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\x03\"\x86\x03\n\rFunctionProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05input\x18\x04 \x03(\t\x12\x0e\n\x06output\x18\x05 \x03(\t\x12\x11\n\tattribute\x18\x06 \x03(\t\x12-\n\x0f\x61ttribute_proto\x18\x0b \x03(\x0b\x32\x14.onnx.AttributeProto\x12\x1d\n\x04node\x18\x07 \x03(\x0b\x32\x0f.onnx.NodeProto\x12\x12\n\ndoc_string\x18\x08 \x01(\t\x12.\n\x0copset_import\x18\t \x03(\x0b\x32\x18.onnx.OperatorSetIdProto\x12\x0e\n\x06\x64omain\x18\n \x01(\t\x12\x10\n\x08overload\x18\r \x01(\t\x12(\n\nvalue_info\x18\x0c \x03(\x0b\x32\x14.onnx.ValueInfoProto\x12\x34\n\x0emetadata_props\x18\x0e \x03(\x0b\x32\x1c.onnx.StringStringEntryProtoJ\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04R\rsince_versionR\x06status*\x97\x02\n\x07Version\x12\x12\n\x0e_START_VERSION\x10\x00\x12\x19\n\x15IR_VERSION_2017_10_10\x10\x01\x12\x19\n\x15IR_VERSION_2017_10_30\x10\x02\x12\x18\n\x14IR_VERSION_2017_11_3\x10\x03\x12\x18\n\x14IR_VERSION_2019_1_22\x10\x04\x12\x18\n\x14IR_VERSION_2019_3_18\x10\x05\x12\x18\n\x14IR_VERSION_2019_9_19\x10\x06\x12\x17\n\x13IR_VERSION_2020_5_8\x10\x07\x12\x18\n\x14IR_VERSION_2021_7_30\x10\x08\x12\x17\n\x13IR_VERSION_2023_5_5\x10\t\x12\x0e\n\nIR_VERSION\x10\n*.\n\x0eOperatorStatus\x12\x10\n\x0c\x45XPERIMENTAL\x10\x00\x12\n\n\x06STABLE\x10\x01\x42\x02H\x03')
 
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'onnx.onnx_ml_pb2', globals())
@@ -31,58 +31,58 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _TENSORPROTO.fields_by_name['double_data']._serialized_options = b'\020\001'
   _TENSORPROTO.fields_by_name['uint64_data']._options = None
   _TENSORPROTO.fields_by_name['uint64_data']._serialized_options = b'\020\001'
-  _VERSION._serialized_start=4192
-  _VERSION._serialized_end=4446
-  _OPERATORSTATUS._serialized_start=4448
-  _OPERATORSTATUS._serialized_end=4494
+  _VERSION._serialized_start=4577
+  _VERSION._serialized_end=4856
+  _OPERATORSTATUS._serialized_start=4858
+  _OPERATORSTATUS._serialized_end=4904
   _ATTRIBUTEPROTO._serialized_start=29
-  _ATTRIBUTEPROTO._serialized_end=745
+  _ATTRIBUTEPROTO._serialized_end=760
   _ATTRIBUTEPROTO_ATTRIBUTETYPE._serialized_start=528
   _ATTRIBUTEPROTO_ATTRIBUTETYPE._serialized_end=745
-  _VALUEINFOPROTO._serialized_start=747
-  _VALUEINFOPROTO._serialized_end=828
-  _NODEPROTO._serialized_start=831
-  _NODEPROTO._serialized_end=981
-  _TRAININGINFOPROTO._serialized_start=984
-  _TRAININGINFOPROTO._serialized_end=1198
-  _MODELPROTO._serialized_start=1201
-  _MODELPROTO._serialized_end=1564
-  _STRINGSTRINGENTRYPROTO._serialized_start=1566
-  _STRINGSTRINGENTRYPROTO._serialized_end=1618
-  _TENSORANNOTATION._serialized_start=1620
-  _TENSORANNOTATION._serialized_end=1727
-  _GRAPHPROTO._serialized_start=1730
-  _GRAPHPROTO._serialized_end=2144
-  _TENSORPROTO._serialized_start=2147
-  _TENSORPROTO._serialized_end=2917
-  _TENSORPROTO_SEGMENT._serialized_start=2542
-  _TENSORPROTO_SEGMENT._serialized_end=2579
-  _TENSORPROTO_DATATYPE._serialized_start=2582
-  _TENSORPROTO_DATATYPE._serialized_end=2874
-  _TENSORPROTO_DATALOCATION._serialized_start=2876
-  _TENSORPROTO_DATALOCATION._serialized_end=2917
-  _SPARSETENSORPROTO._serialized_start=2919
-  _SPARSETENSORPROTO._serialized_end=3023
-  _TENSORSHAPEPROTO._serialized_start=3026
-  _TENSORSHAPEPROTO._serialized_end=3175
-  _TENSORSHAPEPROTO_DIMENSION._serialized_start=3093
-  _TENSORSHAPEPROTO_DIMENSION._serialized_end=3175
-  _TYPEPROTO._serialized_start=3178
-  _TYPEPROTO._serialized_end=3855
-  _TYPEPROTO_TENSOR._serialized_start=3508
-  _TYPEPROTO_TENSOR._serialized_end=3574
-  _TYPEPROTO_SEQUENCE._serialized_start=3576
-  _TYPEPROTO_SEQUENCE._serialized_end=3622
-  _TYPEPROTO_MAP._serialized_start=3624
-  _TYPEPROTO_MAP._serialized_end=3684
-  _TYPEPROTO_OPTIONAL._serialized_start=3686
-  _TYPEPROTO_OPTIONAL._serialized_end=3732
-  _TYPEPROTO_SPARSETENSOR._serialized_start=3734
-  _TYPEPROTO_SPARSETENSOR._serialized_end=3806
-  _TYPEPROTO_OPAQUE._serialized_start=3808
-  _TYPEPROTO_OPAQUE._serialized_end=3846
-  _OPERATORSETIDPROTO._serialized_start=3857
-  _OPERATORSETIDPROTO._serialized_end=3910
-  _FUNCTIONPROTO._serialized_start=3913
-  _FUNCTIONPROTO._serialized_end=4189
+  _VALUEINFOPROTO._serialized_start=763
+  _VALUEINFOPROTO._serialized_end=898
+  _NODEPROTO._serialized_start=901
+  _NODEPROTO._serialized_end=1123
+  _TRAININGINFOPROTO._serialized_start=1126
+  _TRAININGINFOPROTO._serialized_end=1340
+  _MODELPROTO._serialized_start=1343
+  _MODELPROTO._serialized_end=1706
+  _STRINGSTRINGENTRYPROTO._serialized_start=1708
+  _STRINGSTRINGENTRYPROTO._serialized_end=1760
+  _TENSORANNOTATION._serialized_start=1762
+  _TENSORANNOTATION._serialized_end=1869
+  _GRAPHPROTO._serialized_start=1872
+  _GRAPHPROTO._serialized_end=2340
+  _TENSORPROTO._serialized_start=2343
+  _TENSORPROTO._serialized_end=3188
+  _TENSORPROTO_SEGMENT._serialized_start=2792
+  _TENSORPROTO_SEGMENT._serialized_end=2829
+  _TENSORPROTO_DATATYPE._serialized_start=2832
+  _TENSORPROTO_DATATYPE._serialized_end=3145
+  _TENSORPROTO_DATALOCATION._serialized_start=3147
+  _TENSORPROTO_DATALOCATION._serialized_end=3188
+  _SPARSETENSORPROTO._serialized_start=3190
+  _SPARSETENSORPROTO._serialized_end=3294
+  _TENSORSHAPEPROTO._serialized_start=3297
+  _TENSORSHAPEPROTO._serialized_end=3446
+  _TENSORSHAPEPROTO_DIMENSION._serialized_start=3364
+  _TENSORSHAPEPROTO_DIMENSION._serialized_end=3446
+  _TYPEPROTO._serialized_start=3449
+  _TYPEPROTO._serialized_end=4126
+  _TYPEPROTO_TENSOR._serialized_start=3779
+  _TYPEPROTO_TENSOR._serialized_end=3845
+  _TYPEPROTO_SEQUENCE._serialized_start=3847
+  _TYPEPROTO_SEQUENCE._serialized_end=3893
+  _TYPEPROTO_MAP._serialized_start=3895
+  _TYPEPROTO_MAP._serialized_end=3955
+  _TYPEPROTO_OPTIONAL._serialized_start=3957
+  _TYPEPROTO_OPTIONAL._serialized_end=4003
+  _TYPEPROTO_SPARSETENSOR._serialized_start=4005
+  _TYPEPROTO_SPARSETENSOR._serialized_end=4077
+  _TYPEPROTO_OPAQUE._serialized_start=4079
+  _TYPEPROTO_OPAQUE._serialized_end=4117
+  _OPERATORSETIDPROTO._serialized_start=4128
+  _OPERATORSETIDPROTO._serialized_end=4181
+  _FUNCTIONPROTO._serialized_start=4184
+  _FUNCTIONPROTO._serialized_end=4574
 # @@protoc_insertion_point(module_scope)