onnx 1.20.1__cp310-cp310-macosx_12_0_universal2.whl

onnx/onnx_cpp2py_export/__init__.pyi

@@ -0,0 +1,5 @@
+# This is __init__.pyi, not __init__.py
+# This directory is only considered for typing information, not for actual
+# module implementations.
+
+ONNX_ML: bool = ...

onnx/onnx_cpp2py_export/checker.pyi

@@ -0,0 +1,45 @@
+class CheckerContext:
+    ir_version: int = ...
+    opset_imports: dict[str, int] = ...
+
+class LexicalScopeContext:
+    ir_version: int = ...
+    opset_imports: dict[str, int] = ...
+
+class ValidationError(Exception): ...
+
+def check_value_info(bytes: bytes, checker_context: CheckerContext) -> None: ...  # noqa: A002
+def check_tensor(bytes: bytes, checker_context: CheckerContext) -> None: ...  # noqa: A002
+def check_sparse_tensor(bytes: bytes, checker_context: CheckerContext) -> None: ...  # noqa: A002
+def check_attribute(
+    bytes: bytes,  # noqa: A002
+    checker_context: CheckerContext,
+    lexical_scope_context: LexicalScopeContext,
+) -> None: ...
+def check_node(
+    bytes: bytes,  # noqa: A002
+    checker_context: CheckerContext,
+    lexical_scope_context: LexicalScopeContext,
+) -> None: ...
+def check_function(
+    bytes: bytes,  # noqa: A002
+    checker_context: CheckerContext,
+    lexical_scope_context: LexicalScopeContext,
+) -> None: ...
+def check_graph(
+    bytes: bytes,  # noqa: A002
+    checker_context: CheckerContext,
+    lexical_scope_context: LexicalScopeContext,
+) -> None: ...
+def check_model(
+    bytes: bytes,  # noqa: A002
+    full_check: bool,
+    skip_opset_compatibility_check: bool,
+    check_custom_domain: bool,
+) -> None: ...
+def check_model_path(
+    path: str,
+    full_check: bool,
+    skip_opset_compatibility_check: bool,
+    check_custom_domain: bool,
+) -> None: ...

onnx/onnx_cpp2py_export/defs.pyi

@@ -0,0 +1,214 @@
+"""Submodule containing all the ONNX schema definitions."""
+
+from collections.abc import Callable, Sequence
+from typing import overload
+
+from onnx import AttributeProto, FunctionProto
+from onnx.shape_inference import InferenceContext
+
+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: ...
+    def set_type_and_shape_inference_function(
+        self, func: Callable[[InferenceContext], None]
+    ) -> None: ...
+    def get_type_and_shape_inference_function(
+        self,
+    ) -> Callable[[InferenceContext], None]: ...
+
+    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 NodeDeterminism:
+        Deterministic: OpSchema.NodeDeterminism = ...
+        NonDeterministic: OpSchema.NodeDeterminism = ...
+        Unknown: OpSchema.NodeDeterminism = ...
+
+    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,
+            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,26 @@
+# Copyright (c) ONNX Project Contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+
+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.
+    """
+
+def inline_selected_functions2(
+    model: bytes, function_ids: list[tuple[str, str]], exclude: bool
+) -> bytes:
+    """Inlines calls to selected model-local + schema 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,27 @@
+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,45 @@
+from onnx.onnx_pb import (
+    AttributeProto,
+    SparseTensorProto,
+    TensorProto,
+    TensorShapeProto,
+    TypeProto,
+)
+
+class InferenceError(Exception): ...
+
+class GraphInferencer:
+    def do_inferencing(
+        self, input_types: list[TypeProto], input_data: list[TensorProto | None]
+    ) -> list[TypeProto]: ...
+
+class InferenceContext:
+    def get_attribute(self, name: str) -> AttributeProto: ...
+    def get_num_inputs(self) -> int: ...
+    def get_input_type(self, idx: int) -> TypeProto: ...
+    def has_input(self, idx: int) -> bool: ...
+    def get_input_data(self, idx: int) -> TensorProto: ...
+    def get_num_outputs(self) -> int: ...
+    def get_output_type(self, idx: int) -> TypeProto: ...
+    def set_output_type(self, idx: int, type_proto: TypeProto) -> bool: ...
+    def has_output(self, idx: int) -> bool: ...
+    def get_graph_attribute_inferencer(self, attr_name: str) -> GraphInferencer: ...
+    def get_input_sparse_data(self, idx: int) -> SparseTensorProto: ...
+    def get_symbolic_input(self, idx: int) -> TensorShapeProto: ...
+    def get_display_name(self) -> str: ...
+
+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(
+    b: bytes,
+    input_types: list[bytes],
+    attributes: list[bytes],
+) -> list[bytes]: ...

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_pb.py

@@ -0,0 +1,4 @@
+# This file is generated by setup.py. DO NOT EDIT!
+
+
+from .onnx_data_pb2 import *  # noqa

onnx/onnx_data_pb2.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: onnx/onnx-data.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from onnx import onnx_ml_pb2 as onnx_dot_onnx__ml__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x14onnx/onnx-data.proto\x12\x04onnx\x1a\x12onnx/onnx-ml.proto\"\xf0\x02\n\rSequenceProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\telem_type\x18\x02 \x01(\x05\x12(\n\rtensor_values\x18\x03 \x03(\x0b\x32\x11.onnx.TensorProto\x12\x35\n\x14sparse_tensor_values\x18\x04 \x03(\x0b\x32\x17.onnx.SparseTensorProto\x12,\n\x0fsequence_values\x18\x05 \x03(\x0b\x32\x13.onnx.SequenceProto\x12\"\n\nmap_values\x18\x06 \x03(\x0b\x32\x0e.onnx.MapProto\x12,\n\x0foptional_values\x18\x07 \x03(\x0b\x32\x13.onnx.OptionalProto\"]\n\x08\x44\x61taType\x12\r\n\tUNDEFINED\x10\x00\x12\n\n\x06TENSOR\x10\x01\x12\x11\n\rSPARSE_TENSOR\x10\x02\x12\x0c\n\x08SEQUENCE\x10\x03\x12\x07\n\x03MAP\x10\x04\x12\x0c\n\x08OPTIONAL\x10\x05\"r\n\x08MapProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08key_type\x18\x02 \x01(\x05\x12\x0c\n\x04keys\x18\x03 \x03(\x03\x12\x13\n\x0bstring_keys\x18\x04 \x03(\x0c\x12#\n\x06values\x18\x05 \x01(\x0b\x32\x13.onnx.SequenceProto\"\xeb\x02\n\rOptionalProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\telem_type\x18\x02 \x01(\x05\x12\'\n\x0ctensor_value\x18\x03 \x01(\x0b\x32\x11.onnx.TensorProto\x12\x34\n\x13sparse_tensor_value\x18\x04 \x01(\x0b\x32\x17.onnx.SparseTensorProto\x12+\n\x0esequence_value\x18\x05 \x01(\x0b\x32\x13.onnx.SequenceProto\x12!\n\tmap_value\x18\x06 \x01(\x0b\x32\x0e.onnx.MapProto\x12+\n\x0eoptional_value\x18\x07 \x01(\x0b\x32\x13.onnx.OptionalProto\"]\n\x08\x44\x61taType\x12\r\n\tUNDEFINED\x10\x00\x12\n\n\x06TENSOR\x10\x01\x12\x11\n\rSPARSE_TENSOR\x10\x02\x12\x0c\n\x08SEQUENCE\x10\x03\x12\x07\n\x03MAP\x10\x04\x12\x0c\n\x08OPTIONAL\x10\x05\x42\x02H\x03')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'onnx.onnx_data_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'H\003'
+  _globals['_SEQUENCEPROTO']._serialized_start=51
+  _globals['_SEQUENCEPROTO']._serialized_end=419
+  _globals['_SEQUENCEPROTO_DATATYPE']._serialized_start=326
+  _globals['_SEQUENCEPROTO_DATATYPE']._serialized_end=419
+  _globals['_MAPPROTO']._serialized_start=421
+  _globals['_MAPPROTO']._serialized_end=535
+  _globals['_OPTIONALPROTO']._serialized_start=538
+  _globals['_OPTIONALPROTO']._serialized_end=901
+  _globals['_OPTIONALPROTO_DATATYPE']._serialized_start=326
+  _globals['_OPTIONALPROTO_DATATYPE']._serialized_end=419
+# @@protoc_insertion_point(module_scope)

onnx/onnx_data_pb2.pyi

@@ -0,0 +1,86 @@
+from onnx import onnx_ml_pb2 as _onnx_ml_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class SequenceProto(_message.Message):
+    __slots__ = ("name", "elem_type", "tensor_values", "sparse_tensor_values", "sequence_values", "map_values", "optional_values")
+    class DataType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        UNDEFINED: _ClassVar[SequenceProto.DataType]
+        TENSOR: _ClassVar[SequenceProto.DataType]
+        SPARSE_TENSOR: _ClassVar[SequenceProto.DataType]
+        SEQUENCE: _ClassVar[SequenceProto.DataType]
+        MAP: _ClassVar[SequenceProto.DataType]
+        OPTIONAL: _ClassVar[SequenceProto.DataType]
+    UNDEFINED: SequenceProto.DataType
+    TENSOR: SequenceProto.DataType
+    SPARSE_TENSOR: SequenceProto.DataType
+    SEQUENCE: SequenceProto.DataType
+    MAP: SequenceProto.DataType
+    OPTIONAL: SequenceProto.DataType
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    ELEM_TYPE_FIELD_NUMBER: _ClassVar[int]
+    TENSOR_VALUES_FIELD_NUMBER: _ClassVar[int]
+    SPARSE_TENSOR_VALUES_FIELD_NUMBER: _ClassVar[int]
+    SEQUENCE_VALUES_FIELD_NUMBER: _ClassVar[int]
+    MAP_VALUES_FIELD_NUMBER: _ClassVar[int]
+    OPTIONAL_VALUES_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    elem_type: int
+    tensor_values: _containers.RepeatedCompositeFieldContainer[_onnx_ml_pb2.TensorProto]
+    sparse_tensor_values: _containers.RepeatedCompositeFieldContainer[_onnx_ml_pb2.SparseTensorProto]
+    sequence_values: _containers.RepeatedCompositeFieldContainer[SequenceProto]
+    map_values: _containers.RepeatedCompositeFieldContainer[MapProto]
+    optional_values: _containers.RepeatedCompositeFieldContainer[OptionalProto]
+    def __init__(self, name: _Optional[str] = ..., elem_type: _Optional[int] = ..., tensor_values: _Optional[_Iterable[_Union[_onnx_ml_pb2.TensorProto, _Mapping]]] = ..., sparse_tensor_values: _Optional[_Iterable[_Union[_onnx_ml_pb2.SparseTensorProto, _Mapping]]] = ..., sequence_values: _Optional[_Iterable[_Union[SequenceProto, _Mapping]]] = ..., map_values: _Optional[_Iterable[_Union[MapProto, _Mapping]]] = ..., optional_values: _Optional[_Iterable[_Union[OptionalProto, _Mapping]]] = ...) -> None: ...
+
+class MapProto(_message.Message):
+    __slots__ = ("name", "key_type", "keys", "string_keys", "values")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    KEY_TYPE_FIELD_NUMBER: _ClassVar[int]
+    KEYS_FIELD_NUMBER: _ClassVar[int]
+    STRING_KEYS_FIELD_NUMBER: _ClassVar[int]
+    VALUES_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    key_type: int
+    keys: _containers.RepeatedScalarFieldContainer[int]
+    string_keys: _containers.RepeatedScalarFieldContainer[bytes]
+    values: SequenceProto
+    def __init__(self, name: _Optional[str] = ..., key_type: _Optional[int] = ..., keys: _Optional[_Iterable[int]] = ..., string_keys: _Optional[_Iterable[bytes]] = ..., values: _Optional[_Union[SequenceProto, _Mapping]] = ...) -> None: ...
+
+class OptionalProto(_message.Message):
+    __slots__ = ("name", "elem_type", "tensor_value", "sparse_tensor_value", "sequence_value", "map_value", "optional_value")
+    class DataType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        UNDEFINED: _ClassVar[OptionalProto.DataType]
+        TENSOR: _ClassVar[OptionalProto.DataType]
+        SPARSE_TENSOR: _ClassVar[OptionalProto.DataType]
+        SEQUENCE: _ClassVar[OptionalProto.DataType]
+        MAP: _ClassVar[OptionalProto.DataType]
+        OPTIONAL: _ClassVar[OptionalProto.DataType]
+    UNDEFINED: OptionalProto.DataType
+    TENSOR: OptionalProto.DataType
+    SPARSE_TENSOR: OptionalProto.DataType
+    SEQUENCE: OptionalProto.DataType
+    MAP: OptionalProto.DataType
+    OPTIONAL: OptionalProto.DataType
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    ELEM_TYPE_FIELD_NUMBER: _ClassVar[int]
+    TENSOR_VALUE_FIELD_NUMBER: _ClassVar[int]
+    SPARSE_TENSOR_VALUE_FIELD_NUMBER: _ClassVar[int]
+    SEQUENCE_VALUE_FIELD_NUMBER: _ClassVar[int]
+    MAP_VALUE_FIELD_NUMBER: _ClassVar[int]
+    OPTIONAL_VALUE_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    elem_type: int
+    tensor_value: _onnx_ml_pb2.TensorProto
+    sparse_tensor_value: _onnx_ml_pb2.SparseTensorProto
+    sequence_value: SequenceProto
+    map_value: MapProto
+    optional_value: OptionalProto
+    def __init__(self, name: _Optional[str] = ..., elem_type: _Optional[int] = ..., tensor_value: _Optional[_Union[_onnx_ml_pb2.TensorProto, _Mapping]] = ..., sparse_tensor_value: _Optional[_Union[_onnx_ml_pb2.SparseTensorProto, _Mapping]] = ..., sequence_value: _Optional[_Union[SequenceProto, _Mapping]] = ..., map_value: _Optional[_Union[MapProto, _Mapping]] = ..., optional_value: _Optional[_Union[OptionalProto, _Mapping]] = ...) -> None: ...

onnx/onnx_ml_pb2.py

@@ -0,0 +1,101 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: onnx/onnx-ml.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+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\"\xa1\x02\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\x12\x41\n\x15\x64\x65vice_configurations\x18\n \x03(\x0b\x32\".onnx.NodeDeviceConfigurationProto\"2\n\x14IntIntListEntryProto\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\r\n\x05value\x18\x02 \x03(\x03\"\x80\x01\n\x1cNodeDeviceConfigurationProto\x12\x18\n\x10\x63onfiguration_id\x18\x01 \x01(\t\x12.\n\rsharding_spec\x18\x02 \x03(\x0b\x32\x17.onnx.ShardingSpecProto\x12\x16\n\x0epipeline_stage\x18\x03 \x01(\x05\"\xa3\x01\n\x11ShardingSpecProto\x12\x13\n\x0btensor_name\x18\x01 \x01(\t\x12\x0e\n\x06\x64\x65vice\x18\x02 \x03(\x03\x12=\n\x19index_to_device_group_map\x18\x03 \x03(\x0b\x32\x1a.onnx.IntIntListEntryProto\x12*\n\x0bsharded_dim\x18\x04 \x03(\x0b\x32\x15.onnx.ShardedDimProto\"U\n\x0fShardedDimProto\x12\x0c\n\x04\x61xis\x18\x01 \x01(\x03\x12\x34\n\x0fsimple_sharding\x18\x02 \x03(\x0b\x32\x1b.onnx.SimpleShardedDimProto\"\\\n\x15SimpleShardedDimProto\x12\x13\n\tdim_value\x18\x01 \x01(\x03H\x00\x12\x13\n\tdim_param\x18\x02 \x01(\tH\x00\x12\x12\n\nnum_shards\x18\x03 \x01(\x03\x42\x05\n\x03\x64im\"\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\"\xa2\x03\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\x12\x35\n\rconfiguration\x18\x1a \x03(\x0b\x32\x1e.onnx.DeviceConfigurationProto\"M\n\x18\x44\x65viceConfigurationProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0bnum_devices\x18\x02 \x01(\x05\x12\x0e\n\x06\x64\x65vice\x18\x03 \x03(\t\"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\"\x82\x07\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\"\xee\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\x12\x0e\n\nFLOAT4E2M1\x10\x17\x12\x0e\n\nFLOAT8E8M0\x10\x18\x12\t\n\x05UINT2\x10\x19\x12\x08\n\x04INT2\x10\x1a\")\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*\xe7\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\x18\n\x14IR_VERSION_2024_3_25\x10\n\x12\x19\n\x15IR_VERSION_2025_05_12\x10\x0b\x12\x19\n\x15IR_VERSION_2025_08_26\x10\x0c\x12\x0e\n\nIR_VERSION\x10\r*.\n\x0eOperatorStatus\x12\x10\n\x0c\x45XPERIMENTAL\x10\x00\x12\n\n\x06STABLE\x10\x01\x42\x02H\x03')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'onnx.onnx_ml_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'H\003'
+  _globals['_TENSORPROTO'].fields_by_name['float_data']._options = None
+  _globals['_TENSORPROTO'].fields_by_name['float_data']._serialized_options = b'\020\001'
+  _globals['_TENSORPROTO'].fields_by_name['int32_data']._options = None
+  _globals['_TENSORPROTO'].fields_by_name['int32_data']._serialized_options = b'\020\001'
+  _globals['_TENSORPROTO'].fields_by_name['int64_data']._options = None
+  _globals['_TENSORPROTO'].fields_by_name['int64_data']._serialized_options = b'\020\001'
+  _globals['_TENSORPROTO'].fields_by_name['double_data']._options = None
+  _globals['_TENSORPROTO'].fields_by_name['double_data']._serialized_options = b'\020\001'
+  _globals['_TENSORPROTO'].fields_by_name['uint64_data']._options = None
+  _globals['_TENSORPROTO'].fields_by_name['uint64_data']._serialized_options = b'\020\001'
+  _globals['_VERSION']._serialized_start=5361
+  _globals['_VERSION']._serialized_end=5720
+  _globals['_OPERATORSTATUS']._serialized_start=5722
+  _globals['_OPERATORSTATUS']._serialized_end=5768
+  _globals['_ATTRIBUTEPROTO']._serialized_start=29
+  _globals['_ATTRIBUTEPROTO']._serialized_end=760
+  _globals['_ATTRIBUTEPROTO_ATTRIBUTETYPE']._serialized_start=528
+  _globals['_ATTRIBUTEPROTO_ATTRIBUTETYPE']._serialized_end=745
+  _globals['_VALUEINFOPROTO']._serialized_start=763
+  _globals['_VALUEINFOPROTO']._serialized_end=898
+  _globals['_NODEPROTO']._serialized_start=901
+  _globals['_NODEPROTO']._serialized_end=1190
+  _globals['_INTINTLISTENTRYPROTO']._serialized_start=1192
+  _globals['_INTINTLISTENTRYPROTO']._serialized_end=1242
+  _globals['_NODEDEVICECONFIGURATIONPROTO']._serialized_start=1245
+  _globals['_NODEDEVICECONFIGURATIONPROTO']._serialized_end=1373
+  _globals['_SHARDINGSPECPROTO']._serialized_start=1376
+  _globals['_SHARDINGSPECPROTO']._serialized_end=1539
+  _globals['_SHARDEDDIMPROTO']._serialized_start=1541
+  _globals['_SHARDEDDIMPROTO']._serialized_end=1626
+  _globals['_SIMPLESHARDEDDIMPROTO']._serialized_start=1628
+  _globals['_SIMPLESHARDEDDIMPROTO']._serialized_end=1720
+  _globals['_TRAININGINFOPROTO']._serialized_start=1723
+  _globals['_TRAININGINFOPROTO']._serialized_end=1937
+  _globals['_MODELPROTO']._serialized_start=1940
+  _globals['_MODELPROTO']._serialized_end=2358
+  _globals['_DEVICECONFIGURATIONPROTO']._serialized_start=2360
+  _globals['_DEVICECONFIGURATIONPROTO']._serialized_end=2437
+  _globals['_STRINGSTRINGENTRYPROTO']._serialized_start=2439
+  _globals['_STRINGSTRINGENTRYPROTO']._serialized_end=2491
+  _globals['_TENSORANNOTATION']._serialized_start=2493
+  _globals['_TENSORANNOTATION']._serialized_end=2600
+  _globals['_GRAPHPROTO']._serialized_start=2603
+  _globals['_GRAPHPROTO']._serialized_end=3071
+  _globals['_TENSORPROTO']._serialized_start=3074
+  _globals['_TENSORPROTO']._serialized_end=3972
+  _globals['_TENSORPROTO_SEGMENT']._serialized_start=3523
+  _globals['_TENSORPROTO_SEGMENT']._serialized_end=3560
+  _globals['_TENSORPROTO_DATATYPE']._serialized_start=3563
+  _globals['_TENSORPROTO_DATATYPE']._serialized_end=3929
+  _globals['_TENSORPROTO_DATALOCATION']._serialized_start=3931
+  _globals['_TENSORPROTO_DATALOCATION']._serialized_end=3972
+  _globals['_SPARSETENSORPROTO']._serialized_start=3974
+  _globals['_SPARSETENSORPROTO']._serialized_end=4078
+  _globals['_TENSORSHAPEPROTO']._serialized_start=4081
+  _globals['_TENSORSHAPEPROTO']._serialized_end=4230
+  _globals['_TENSORSHAPEPROTO_DIMENSION']._serialized_start=4148
+  _globals['_TENSORSHAPEPROTO_DIMENSION']._serialized_end=4230
+  _globals['_TYPEPROTO']._serialized_start=4233
+  _globals['_TYPEPROTO']._serialized_end=4910
+  _globals['_TYPEPROTO_TENSOR']._serialized_start=4563
+  _globals['_TYPEPROTO_TENSOR']._serialized_end=4629
+  _globals['_TYPEPROTO_SEQUENCE']._serialized_start=4631
+  _globals['_TYPEPROTO_SEQUENCE']._serialized_end=4677
+  _globals['_TYPEPROTO_MAP']._serialized_start=4679
+  _globals['_TYPEPROTO_MAP']._serialized_end=4739
+  _globals['_TYPEPROTO_OPTIONAL']._serialized_start=4741
+  _globals['_TYPEPROTO_OPTIONAL']._serialized_end=4787
+  _globals['_TYPEPROTO_SPARSETENSOR']._serialized_start=4789
+  _globals['_TYPEPROTO_SPARSETENSOR']._serialized_end=4861
+  _globals['_TYPEPROTO_OPAQUE']._serialized_start=4863
+  _globals['_TYPEPROTO_OPAQUE']._serialized_end=4901
+  _globals['_OPERATORSETIDPROTO']._serialized_start=4912
+  _globals['_OPERATORSETIDPROTO']._serialized_end=4965
+  _globals['_FUNCTIONPROTO']._serialized_start=4968
+  _globals['_FUNCTIONPROTO']._serialized_end=5358
+# @@protoc_insertion_point(module_scope)