1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
|
"""
@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
|
