""" @generated by mypy-protobuf. Do not edit manually! isort:skip_file """ import builtins import caffe2.proto.caffe2_pb2 import google.protobuf.descriptor import google.protobuf.internal.containers import google.protobuf.internal.enum_type_wrapper import google.protobuf.message import typing import typing_extensions DESCRIPTOR: google.protobuf.descriptor.FileDescriptor = ... global___ProtoVersion = ProtoVersion class _ProtoVersion(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ProtoVersion], type): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor = ... PROTO_VERSION_NEWEST = ProtoVersion.V(6) class ProtoVersion(metaclass=_ProtoVersion): V = typing.NewType('V', int) PROTO_VERSION_NEWEST = ProtoVersion.V(6) class RecordRef(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor = ... KEY_FIELD_NUMBER: int key: typing.Text = ... def __init__(self, *, key : typing.Optional[typing.Text] = ..., ) -> None: ... def HasField(self, field_name: typing_extensions.Literal[u"key",b"key"]) -> bool: ... def ClearField(self, field_name: typing_extensions.Literal[u"key",b"key"]) -> None: ... global___RecordRef = RecordRef class TensorDef(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor = ... DIMS_FIELD_NUMBER: int OFFSET_FIELD_NUMBER: int STRIDES_FIELD_NUMBER: int REQUIRES_GRAD_FIELD_NUMBER: int DATA_TYPE_FIELD_NUMBER: int DATA_FIELD_NUMBER: int DEVICE_FIELD_NUMBER: int IS_QUANTIZED_FIELD_NUMBER: int SCALE_FIELD_NUMBER: int ZERO_POINT_FIELD_NUMBER: int dims: google.protobuf.internal.containers.RepeatedScalarFieldContainer[int] = ... offset: int = ... strides: google.protobuf.internal.containers.RepeatedScalarFieldContainer[int] = ... requires_grad: bool = ... data_type: caffe2.proto.caffe2_pb2.TensorProto.DataType = ... device: typing.Text = ... is_quantized: bool = ... scale: float = ... zero_point: int = ... @property def data(self) -> global___RecordRef: ... def __init__(self, *, dims : typing.Optional[typing.Iterable[int]] = ..., offset : typing.Optional[int] = ..., strides : typing.Optional[typing.Iterable[int]] = ..., requires_grad : typing.Optional[bool] = ..., data_type : typing.Optional[caffe2.proto.caffe2_pb2.TensorProto.DataType] = ..., data : typing.Optional[global___RecordRef] = ..., device : typing.Optional[typing.Text] = ..., is_quantized : typing.Optional[bool] = ..., scale : typing.Optional[float] = ..., zero_point : typing.Optional[int] = ..., ) -> None: ... def HasField(self, field_name: typing_extensions.Literal[u"data",b"data",u"data_type",b"data_type",u"device",b"device",u"is_quantized",b"is_quantized",u"offset",b"offset",u"requires_grad",b"requires_grad",u"scale",b"scale",u"zero_point",b"zero_point"]) -> bool: ... def ClearField(self, field_name: typing_extensions.Literal[u"data",b"data",u"data_type",b"data_type",u"device",b"device",u"dims",b"dims",u"is_quantized",b"is_quantized",u"offset",b"offset",u"requires_grad",b"requires_grad",u"scale",b"scale",u"strides",b"strides",u"zero_point",b"zero_point"]) -> None: ... global___TensorDef = TensorDef class AttributeDef(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor = ... TYPE_FIELD_NUMBER: int NAME_FIELD_NUMBER: int ID_FIELD_NUMBER: int type: typing.Text = ... name: typing.Text = ... id: int = ... def __init__(self, *, type : typing.Optional[typing.Text] = ..., name : typing.Optional[typing.Text] = ..., id : typing.Optional[int] = ..., ) -> None: ... def HasField(self, field_name: typing_extensions.Literal[u"id",b"id",u"name",b"name",u"type",b"type"]) -> bool: ... def ClearField(self, field_name: typing_extensions.Literal[u"id",b"id",u"name",b"name",u"type",b"type"]) -> None: ... global___AttributeDef = AttributeDef class ParameterDef(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor = ... IS_BUFFER_FIELD_NUMBER: int TENSOR_ID_FIELD_NUMBER: int NAME_FIELD_NUMBER: int is_buffer: bool = ... tensor_id: int = ... name: typing.Text = ... def __init__(self, *, is_buffer : typing.Optional[bool] = ..., tensor_id : typing.Optional[int] = ..., name : typing.Optional[typing.Text] = ..., ) -> None: ... def HasField(self, field_name: typing_extensions.Literal[u"is_buffer",b"is_buffer",u"name",b"name",u"tensor_id",b"tensor_id"]) -> bool: ... def ClearField(self, field_name: typing_extensions.Literal[u"is_buffer",b"is_buffer",u"name",b"name",u"tensor_id",b"tensor_id"]) -> None: ... global___ParameterDef = ParameterDef class ModuleDef(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor = ... SUBMODULES_FIELD_NUMBER: int TORCHSCRIPT_ARENA_FIELD_NUMBER: int CAFFE2_NETS_FIELD_NUMBER: int PICKLE_ARENA_FIELD_NUMBER: int CPP_ARENA_FIELD_NUMBER: int PARAMETERS_FIELD_NUMBER: int NAME_FIELD_NUMBER: int OPTIMIZE_FIELD_NUMBER: int ATTRIBUTES_FIELD_NUMBER: int GET_STATE_ATTRIBUTE_ID_FIELD_NUMBER: int TORCHSCRIPT_DEBUG_ARENA_FIELD_NUMBER: int name: typing.Text = ... optimize: bool = ... get_state_attribute_id: int = ... @property def submodules(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ModuleDef]: ... @property def torchscript_arena(self) -> global___RecordRef: ... @property def caffe2_nets(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[caffe2.proto.caffe2_pb2.NetDef]: ... @property def pickle_arena(self) -> global___RecordRef: ... @property def cpp_arena(self) -> global___RecordRef: ... @property def parameters(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ParameterDef]: ... @property def attributes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___AttributeDef]: ... @property def torchscript_debug_arena(self) -> global___RecordRef: ... def __init__(self, *, submodules : typing.Optional[typing.Iterable[global___ModuleDef]] = ..., torchscript_arena : typing.Optional[global___RecordRef] = ..., caffe2_nets : typing.Optional[typing.Iterable[caffe2.proto.caffe2_pb2.NetDef]] = ..., pickle_arena : typing.Optional[global___RecordRef] = ..., cpp_arena : typing.Optional[global___RecordRef] = ..., parameters : typing.Optional[typing.Iterable[global___ParameterDef]] = ..., name : typing.Optional[typing.Text] = ..., optimize : typing.Optional[bool] = ..., attributes : typing.Optional[typing.Iterable[global___AttributeDef]] = ..., get_state_attribute_id : typing.Optional[int] = ..., torchscript_debug_arena : typing.Optional[global___RecordRef] = ..., ) -> None: ... def HasField(self, field_name: typing_extensions.Literal[u"cpp_arena",b"cpp_arena",u"get_state_attribute_id",b"get_state_attribute_id",u"name",b"name",u"optimize",b"optimize",u"pickle_arena",b"pickle_arena",u"torchscript_arena",b"torchscript_arena",u"torchscript_debug_arena",b"torchscript_debug_arena"]) -> bool: ... def ClearField(self, field_name: typing_extensions.Literal[u"attributes",b"attributes",u"caffe2_nets",b"caffe2_nets",u"cpp_arena",b"cpp_arena",u"get_state_attribute_id",b"get_state_attribute_id",u"name",b"name",u"optimize",b"optimize",u"parameters",b"parameters",u"pickle_arena",b"pickle_arena",u"submodules",b"submodules",u"torchscript_arena",b"torchscript_arena",u"torchscript_debug_arena",b"torchscript_debug_arena"]) -> None: ... global___ModuleDef = ModuleDef class LibDef(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor = ... TORCHSCRIPT_ARENA_FIELD_NUMBER: int @property def torchscript_arena(self) -> global___RecordRef: ... def __init__(self, *, torchscript_arena : typing.Optional[global___RecordRef] = ..., ) -> None: ... def HasField(self, field_name: typing_extensions.Literal[u"torchscript_arena",b"torchscript_arena"]) -> bool: ... def ClearField(self, field_name: typing_extensions.Literal[u"torchscript_arena",b"torchscript_arena"]) -> None: ... global___LibDef = LibDef class ModelDef(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor = ... PROTO_VERSION_FIELD_NUMBER: int MAIN_MODULE_FIELD_NUMBER: int PRODUCER_NAME_FIELD_NUMBER: int PRODUCER_VERSION_FIELD_NUMBER: int TENSORS_FIELD_NUMBER: int proto_version: int = ... producer_name: typing.Text = ... producer_version: typing.Text = ... @property def main_module(self) -> global___ModuleDef: ... @property def tensors(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TensorDef]: ... def __init__(self, *, proto_version : typing.Optional[int] = ..., main_module : typing.Optional[global___ModuleDef] = ..., producer_name : typing.Optional[typing.Text] = ..., producer_version : typing.Optional[typing.Text] = ..., tensors : typing.Optional[typing.Iterable[global___TensorDef]] = ..., ) -> None: ... def HasField(self, field_name: typing_extensions.Literal[u"main_module",b"main_module",u"producer_name",b"producer_name",u"producer_version",b"producer_version",u"proto_version",b"proto_version"]) -> bool: ... def ClearField(self, field_name: typing_extensions.Literal[u"main_module",b"main_module",u"producer_name",b"producer_name",u"producer_version",b"producer_version",u"proto_version",b"proto_version",u"tensors",b"tensors"]) -> None: ... global___ModelDef = ModelDef