torch_pb2.py 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538
  1. # -*- coding: utf-8 -*-
  2. # Generated by the protocol buffer compiler. DO NOT EDIT!
  3. # source: caffe2/proto/torch.proto
  4. """Generated protocol buffer code."""
  5. from google.protobuf.internal import enum_type_wrapper
  6. from google.protobuf import descriptor as _descriptor
  7. from google.protobuf import message as _message
  8. from google.protobuf import reflection as _reflection
  9. from google.protobuf import symbol_database as _symbol_database
  10. # @@protoc_insertion_point(imports)
  11. _sym_db = _symbol_database.Default()
  12. from caffe2.proto import caffe2_pb2 as caffe2_dot_proto_dot_caffe2__pb2
  13. DESCRIPTOR = _descriptor.FileDescriptor(
  14. name='caffe2/proto/torch.proto',
  15. package='torch',
  16. syntax='proto2',
  17. serialized_options=None,
  18. create_key=_descriptor._internal_create_key,
  19. serialized_pb=b'\n\x18\x63\x61\x66\x66\x65\x32/proto/torch.proto\x12\x05torch\x1a\x19\x63\x61\x66\x66\x65\x32/proto/caffe2.proto\"\x18\n\tRecordRef\x12\x0b\n\x03key\x18\x01 \x01(\t\"\xeb\x01\n\tTensorDef\x12\x0c\n\x04\x64ims\x18\x01 \x03(\x03\x12\x0e\n\x06offset\x18\x02 \x01(\x03\x12\x0f\n\x07strides\x18\x03 \x03(\x03\x12\x15\n\rrequires_grad\x18\x04 \x01(\x08\x12/\n\tdata_type\x18\x05 \x01(\x0e\x32\x1c.caffe2.TensorProto.DataType\x12\x1e\n\x04\x64\x61ta\x18\x06 \x01(\x0b\x32\x10.torch.RecordRef\x12\x0e\n\x06\x64\x65vice\x18\x07 \x01(\t\x12\x14\n\x0cis_quantized\x18\x08 \x01(\x08\x12\r\n\x05scale\x18\t \x01(\x01\x12\x12\n\nzero_point\x18\n \x01(\x03\"6\n\x0c\x41ttributeDef\x12\x0c\n\x04type\x18\x01 \x02(\t\x12\x0c\n\x04name\x18\x02 \x02(\t\x12\n\n\x02id\x18\x03 \x02(\x03\"B\n\x0cParameterDef\x12\x11\n\tis_buffer\x18\x01 \x01(\x08\x12\x11\n\ttensor_id\x18\x02 \x01(\x03\x12\x0c\n\x04name\x18\x03 \x01(\t\"\x95\x03\n\tModuleDef\x12$\n\nsubmodules\x18\x01 \x03(\x0b\x32\x10.torch.ModuleDef\x12+\n\x11torchscript_arena\x18\x02 \x01(\x0b\x32\x10.torch.RecordRef\x12#\n\x0b\x63\x61\x66\x66\x65\x32_nets\x18\x03 \x03(\x0b\x32\x0e.caffe2.NetDef\x12&\n\x0cpickle_arena\x18\x04 \x01(\x0b\x32\x10.torch.RecordRef\x12#\n\tcpp_arena\x18\x05 \x01(\x0b\x32\x10.torch.RecordRef\x12\'\n\nparameters\x18\x06 \x03(\x0b\x32\x13.torch.ParameterDef\x12\x0c\n\x04name\x18\x07 \x01(\t\x12\x10\n\x08optimize\x18\x08 \x01(\x08\x12\'\n\nattributes\x18\t \x03(\x0b\x32\x13.torch.AttributeDef\x12\x1e\n\x16get_state_attribute_id\x18\n \x01(\x03\x12\x31\n\x17torchscript_debug_arena\x18\x0b \x01(\x0b\x32\x10.torch.RecordRef\"5\n\x06LibDef\x12+\n\x11torchscript_arena\x18\x01 \x01(\x0b\x32\x10.torch.RecordRef\"\xa8\x01\n\x08ModelDef\x12\x15\n\rproto_version\x18\x01 \x01(\x03\x12%\n\x0bmain_module\x18\x02 \x01(\x0b\x32\x10.torch.ModuleDef\x12\x15\n\rproducer_name\x18\x03 \x01(\t\x12\x18\n\x10producer_version\x18\x04 \x01(\t\x12!\n\x07tensors\x18\x05 \x03(\x0b\x32\x10.torch.TensorDefJ\x04\x08\t\x10\nR\x04libs*(\n\x0cProtoVersion\x12\x18\n\x14PROTO_VERSION_NEWEST\x10\x06'
  20. ,
  21. dependencies=[caffe2_dot_proto_dot_caffe2__pb2.DESCRIPTOR,])
  22. _PROTOVERSION = _descriptor.EnumDescriptor(
  23. name='ProtoVersion',
  24. full_name='torch.ProtoVersion',
  25. filename=None,
  26. file=DESCRIPTOR,
  27. create_key=_descriptor._internal_create_key,
  28. values=[
  29. _descriptor.EnumValueDescriptor(
  30. name='PROTO_VERSION_NEWEST', index=0, number=6,
  31. serialized_options=None,
  32. type=None,
  33. create_key=_descriptor._internal_create_key),
  34. ],
  35. containing_type=None,
  36. serialized_options=None,
  37. serialized_start=1084,
  38. serialized_end=1124,
  39. )
  40. _sym_db.RegisterEnumDescriptor(_PROTOVERSION)
  41. ProtoVersion = enum_type_wrapper.EnumTypeWrapper(_PROTOVERSION)
  42. PROTO_VERSION_NEWEST = 6
  43. _RECORDREF = _descriptor.Descriptor(
  44. name='RecordRef',
  45. full_name='torch.RecordRef',
  46. filename=None,
  47. file=DESCRIPTOR,
  48. containing_type=None,
  49. create_key=_descriptor._internal_create_key,
  50. fields=[
  51. _descriptor.FieldDescriptor(
  52. name='key', full_name='torch.RecordRef.key', index=0,
  53. number=1, type=9, cpp_type=9, label=1,
  54. has_default_value=False, default_value=b"".decode('utf-8'),
  55. message_type=None, enum_type=None, containing_type=None,
  56. is_extension=False, extension_scope=None,
  57. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  58. ],
  59. extensions=[
  60. ],
  61. nested_types=[],
  62. enum_types=[
  63. ],
  64. serialized_options=None,
  65. is_extendable=False,
  66. syntax='proto2',
  67. extension_ranges=[],
  68. oneofs=[
  69. ],
  70. serialized_start=62,
  71. serialized_end=86,
  72. )
  73. _TENSORDEF = _descriptor.Descriptor(
  74. name='TensorDef',
  75. full_name='torch.TensorDef',
  76. filename=None,
  77. file=DESCRIPTOR,
  78. containing_type=None,
  79. create_key=_descriptor._internal_create_key,
  80. fields=[
  81. _descriptor.FieldDescriptor(
  82. name='dims', full_name='torch.TensorDef.dims', index=0,
  83. number=1, type=3, cpp_type=2, label=3,
  84. has_default_value=False, default_value=[],
  85. message_type=None, enum_type=None, containing_type=None,
  86. is_extension=False, extension_scope=None,
  87. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  88. _descriptor.FieldDescriptor(
  89. name='offset', full_name='torch.TensorDef.offset', index=1,
  90. number=2, type=3, cpp_type=2, label=1,
  91. has_default_value=False, default_value=0,
  92. message_type=None, enum_type=None, containing_type=None,
  93. is_extension=False, extension_scope=None,
  94. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  95. _descriptor.FieldDescriptor(
  96. name='strides', full_name='torch.TensorDef.strides', index=2,
  97. number=3, type=3, cpp_type=2, label=3,
  98. has_default_value=False, default_value=[],
  99. message_type=None, enum_type=None, containing_type=None,
  100. is_extension=False, extension_scope=None,
  101. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  102. _descriptor.FieldDescriptor(
  103. name='requires_grad', full_name='torch.TensorDef.requires_grad', index=3,
  104. number=4, type=8, cpp_type=7, label=1,
  105. has_default_value=False, default_value=False,
  106. message_type=None, enum_type=None, containing_type=None,
  107. is_extension=False, extension_scope=None,
  108. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  109. _descriptor.FieldDescriptor(
  110. name='data_type', full_name='torch.TensorDef.data_type', index=4,
  111. number=5, type=14, cpp_type=8, label=1,
  112. has_default_value=False, default_value=0,
  113. message_type=None, enum_type=None, containing_type=None,
  114. is_extension=False, extension_scope=None,
  115. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  116. _descriptor.FieldDescriptor(
  117. name='data', full_name='torch.TensorDef.data', index=5,
  118. number=6, type=11, cpp_type=10, label=1,
  119. has_default_value=False, default_value=None,
  120. message_type=None, enum_type=None, containing_type=None,
  121. is_extension=False, extension_scope=None,
  122. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  123. _descriptor.FieldDescriptor(
  124. name='device', full_name='torch.TensorDef.device', index=6,
  125. number=7, type=9, cpp_type=9, label=1,
  126. has_default_value=False, default_value=b"".decode('utf-8'),
  127. message_type=None, enum_type=None, containing_type=None,
  128. is_extension=False, extension_scope=None,
  129. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  130. _descriptor.FieldDescriptor(
  131. name='is_quantized', full_name='torch.TensorDef.is_quantized', index=7,
  132. number=8, type=8, cpp_type=7, label=1,
  133. has_default_value=False, default_value=False,
  134. message_type=None, enum_type=None, containing_type=None,
  135. is_extension=False, extension_scope=None,
  136. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  137. _descriptor.FieldDescriptor(
  138. name='scale', full_name='torch.TensorDef.scale', index=8,
  139. number=9, type=1, cpp_type=5, label=1,
  140. has_default_value=False, default_value=float(0),
  141. message_type=None, enum_type=None, containing_type=None,
  142. is_extension=False, extension_scope=None,
  143. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  144. _descriptor.FieldDescriptor(
  145. name='zero_point', full_name='torch.TensorDef.zero_point', index=9,
  146. number=10, type=3, cpp_type=2, label=1,
  147. has_default_value=False, default_value=0,
  148. message_type=None, enum_type=None, containing_type=None,
  149. is_extension=False, extension_scope=None,
  150. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  151. ],
  152. extensions=[
  153. ],
  154. nested_types=[],
  155. enum_types=[
  156. ],
  157. serialized_options=None,
  158. is_extendable=False,
  159. syntax='proto2',
  160. extension_ranges=[],
  161. oneofs=[
  162. ],
  163. serialized_start=89,
  164. serialized_end=324,
  165. )
  166. _ATTRIBUTEDEF = _descriptor.Descriptor(
  167. name='AttributeDef',
  168. full_name='torch.AttributeDef',
  169. filename=None,
  170. file=DESCRIPTOR,
  171. containing_type=None,
  172. create_key=_descriptor._internal_create_key,
  173. fields=[
  174. _descriptor.FieldDescriptor(
  175. name='type', full_name='torch.AttributeDef.type', index=0,
  176. number=1, type=9, cpp_type=9, label=2,
  177. has_default_value=False, default_value=b"".decode('utf-8'),
  178. message_type=None, enum_type=None, containing_type=None,
  179. is_extension=False, extension_scope=None,
  180. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  181. _descriptor.FieldDescriptor(
  182. name='name', full_name='torch.AttributeDef.name', index=1,
  183. number=2, type=9, cpp_type=9, label=2,
  184. has_default_value=False, default_value=b"".decode('utf-8'),
  185. message_type=None, enum_type=None, containing_type=None,
  186. is_extension=False, extension_scope=None,
  187. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  188. _descriptor.FieldDescriptor(
  189. name='id', full_name='torch.AttributeDef.id', index=2,
  190. number=3, type=3, cpp_type=2, label=2,
  191. has_default_value=False, default_value=0,
  192. message_type=None, enum_type=None, containing_type=None,
  193. is_extension=False, extension_scope=None,
  194. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  195. ],
  196. extensions=[
  197. ],
  198. nested_types=[],
  199. enum_types=[
  200. ],
  201. serialized_options=None,
  202. is_extendable=False,
  203. syntax='proto2',
  204. extension_ranges=[],
  205. oneofs=[
  206. ],
  207. serialized_start=326,
  208. serialized_end=380,
  209. )
  210. _PARAMETERDEF = _descriptor.Descriptor(
  211. name='ParameterDef',
  212. full_name='torch.ParameterDef',
  213. filename=None,
  214. file=DESCRIPTOR,
  215. containing_type=None,
  216. create_key=_descriptor._internal_create_key,
  217. fields=[
  218. _descriptor.FieldDescriptor(
  219. name='is_buffer', full_name='torch.ParameterDef.is_buffer', index=0,
  220. number=1, type=8, cpp_type=7, label=1,
  221. has_default_value=False, default_value=False,
  222. message_type=None, enum_type=None, containing_type=None,
  223. is_extension=False, extension_scope=None,
  224. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  225. _descriptor.FieldDescriptor(
  226. name='tensor_id', full_name='torch.ParameterDef.tensor_id', index=1,
  227. number=2, type=3, cpp_type=2, label=1,
  228. has_default_value=False, default_value=0,
  229. message_type=None, enum_type=None, containing_type=None,
  230. is_extension=False, extension_scope=None,
  231. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  232. _descriptor.FieldDescriptor(
  233. name='name', full_name='torch.ParameterDef.name', index=2,
  234. number=3, type=9, cpp_type=9, label=1,
  235. has_default_value=False, default_value=b"".decode('utf-8'),
  236. message_type=None, enum_type=None, containing_type=None,
  237. is_extension=False, extension_scope=None,
  238. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  239. ],
  240. extensions=[
  241. ],
  242. nested_types=[],
  243. enum_types=[
  244. ],
  245. serialized_options=None,
  246. is_extendable=False,
  247. syntax='proto2',
  248. extension_ranges=[],
  249. oneofs=[
  250. ],
  251. serialized_start=382,
  252. serialized_end=448,
  253. )
  254. _MODULEDEF = _descriptor.Descriptor(
  255. name='ModuleDef',
  256. full_name='torch.ModuleDef',
  257. filename=None,
  258. file=DESCRIPTOR,
  259. containing_type=None,
  260. create_key=_descriptor._internal_create_key,
  261. fields=[
  262. _descriptor.FieldDescriptor(
  263. name='submodules', full_name='torch.ModuleDef.submodules', index=0,
  264. number=1, type=11, cpp_type=10, label=3,
  265. has_default_value=False, default_value=[],
  266. message_type=None, enum_type=None, containing_type=None,
  267. is_extension=False, extension_scope=None,
  268. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  269. _descriptor.FieldDescriptor(
  270. name='torchscript_arena', full_name='torch.ModuleDef.torchscript_arena', index=1,
  271. number=2, type=11, cpp_type=10, label=1,
  272. has_default_value=False, default_value=None,
  273. message_type=None, enum_type=None, containing_type=None,
  274. is_extension=False, extension_scope=None,
  275. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  276. _descriptor.FieldDescriptor(
  277. name='caffe2_nets', full_name='torch.ModuleDef.caffe2_nets', index=2,
  278. number=3, type=11, cpp_type=10, label=3,
  279. has_default_value=False, default_value=[],
  280. message_type=None, enum_type=None, containing_type=None,
  281. is_extension=False, extension_scope=None,
  282. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  283. _descriptor.FieldDescriptor(
  284. name='pickle_arena', full_name='torch.ModuleDef.pickle_arena', index=3,
  285. number=4, type=11, cpp_type=10, label=1,
  286. has_default_value=False, default_value=None,
  287. message_type=None, enum_type=None, containing_type=None,
  288. is_extension=False, extension_scope=None,
  289. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  290. _descriptor.FieldDescriptor(
  291. name='cpp_arena', full_name='torch.ModuleDef.cpp_arena', index=4,
  292. number=5, type=11, cpp_type=10, label=1,
  293. has_default_value=False, default_value=None,
  294. message_type=None, enum_type=None, containing_type=None,
  295. is_extension=False, extension_scope=None,
  296. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  297. _descriptor.FieldDescriptor(
  298. name='parameters', full_name='torch.ModuleDef.parameters', index=5,
  299. number=6, type=11, cpp_type=10, label=3,
  300. has_default_value=False, default_value=[],
  301. message_type=None, enum_type=None, containing_type=None,
  302. is_extension=False, extension_scope=None,
  303. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  304. _descriptor.FieldDescriptor(
  305. name='name', full_name='torch.ModuleDef.name', index=6,
  306. number=7, type=9, cpp_type=9, label=1,
  307. has_default_value=False, default_value=b"".decode('utf-8'),
  308. message_type=None, enum_type=None, containing_type=None,
  309. is_extension=False, extension_scope=None,
  310. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  311. _descriptor.FieldDescriptor(
  312. name='optimize', full_name='torch.ModuleDef.optimize', index=7,
  313. number=8, type=8, cpp_type=7, label=1,
  314. has_default_value=False, default_value=False,
  315. message_type=None, enum_type=None, containing_type=None,
  316. is_extension=False, extension_scope=None,
  317. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  318. _descriptor.FieldDescriptor(
  319. name='attributes', full_name='torch.ModuleDef.attributes', index=8,
  320. number=9, type=11, cpp_type=10, label=3,
  321. has_default_value=False, default_value=[],
  322. message_type=None, enum_type=None, containing_type=None,
  323. is_extension=False, extension_scope=None,
  324. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  325. _descriptor.FieldDescriptor(
  326. name='get_state_attribute_id', full_name='torch.ModuleDef.get_state_attribute_id', index=9,
  327. number=10, type=3, cpp_type=2, label=1,
  328. has_default_value=False, default_value=0,
  329. message_type=None, enum_type=None, containing_type=None,
  330. is_extension=False, extension_scope=None,
  331. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  332. _descriptor.FieldDescriptor(
  333. name='torchscript_debug_arena', full_name='torch.ModuleDef.torchscript_debug_arena', index=10,
  334. number=11, type=11, cpp_type=10, label=1,
  335. has_default_value=False, default_value=None,
  336. message_type=None, enum_type=None, containing_type=None,
  337. is_extension=False, extension_scope=None,
  338. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  339. ],
  340. extensions=[
  341. ],
  342. nested_types=[],
  343. enum_types=[
  344. ],
  345. serialized_options=None,
  346. is_extendable=False,
  347. syntax='proto2',
  348. extension_ranges=[],
  349. oneofs=[
  350. ],
  351. serialized_start=451,
  352. serialized_end=856,
  353. )
  354. _LIBDEF = _descriptor.Descriptor(
  355. name='LibDef',
  356. full_name='torch.LibDef',
  357. filename=None,
  358. file=DESCRIPTOR,
  359. containing_type=None,
  360. create_key=_descriptor._internal_create_key,
  361. fields=[
  362. _descriptor.FieldDescriptor(
  363. name='torchscript_arena', full_name='torch.LibDef.torchscript_arena', index=0,
  364. number=1, type=11, cpp_type=10, label=1,
  365. has_default_value=False, default_value=None,
  366. message_type=None, enum_type=None, containing_type=None,
  367. is_extension=False, extension_scope=None,
  368. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  369. ],
  370. extensions=[
  371. ],
  372. nested_types=[],
  373. enum_types=[
  374. ],
  375. serialized_options=None,
  376. is_extendable=False,
  377. syntax='proto2',
  378. extension_ranges=[],
  379. oneofs=[
  380. ],
  381. serialized_start=858,
  382. serialized_end=911,
  383. )
  384. _MODELDEF = _descriptor.Descriptor(
  385. name='ModelDef',
  386. full_name='torch.ModelDef',
  387. filename=None,
  388. file=DESCRIPTOR,
  389. containing_type=None,
  390. create_key=_descriptor._internal_create_key,
  391. fields=[
  392. _descriptor.FieldDescriptor(
  393. name='proto_version', full_name='torch.ModelDef.proto_version', index=0,
  394. number=1, type=3, cpp_type=2, label=1,
  395. has_default_value=False, default_value=0,
  396. message_type=None, enum_type=None, containing_type=None,
  397. is_extension=False, extension_scope=None,
  398. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  399. _descriptor.FieldDescriptor(
  400. name='main_module', full_name='torch.ModelDef.main_module', index=1,
  401. number=2, type=11, cpp_type=10, label=1,
  402. has_default_value=False, default_value=None,
  403. message_type=None, enum_type=None, containing_type=None,
  404. is_extension=False, extension_scope=None,
  405. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  406. _descriptor.FieldDescriptor(
  407. name='producer_name', full_name='torch.ModelDef.producer_name', index=2,
  408. number=3, type=9, cpp_type=9, label=1,
  409. has_default_value=False, default_value=b"".decode('utf-8'),
  410. message_type=None, enum_type=None, containing_type=None,
  411. is_extension=False, extension_scope=None,
  412. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  413. _descriptor.FieldDescriptor(
  414. name='producer_version', full_name='torch.ModelDef.producer_version', index=3,
  415. number=4, type=9, cpp_type=9, label=1,
  416. has_default_value=False, default_value=b"".decode('utf-8'),
  417. message_type=None, enum_type=None, containing_type=None,
  418. is_extension=False, extension_scope=None,
  419. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  420. _descriptor.FieldDescriptor(
  421. name='tensors', full_name='torch.ModelDef.tensors', index=4,
  422. number=5, type=11, cpp_type=10, label=3,
  423. has_default_value=False, default_value=[],
  424. message_type=None, enum_type=None, containing_type=None,
  425. is_extension=False, extension_scope=None,
  426. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  427. ],
  428. extensions=[
  429. ],
  430. nested_types=[],
  431. enum_types=[
  432. ],
  433. serialized_options=None,
  434. is_extendable=False,
  435. syntax='proto2',
  436. extension_ranges=[],
  437. oneofs=[
  438. ],
  439. serialized_start=914,
  440. serialized_end=1082,
  441. )
  442. _TENSORDEF.fields_by_name['data_type'].enum_type = caffe2_dot_proto_dot_caffe2__pb2._TENSORPROTO_DATATYPE
  443. _TENSORDEF.fields_by_name['data'].message_type = _RECORDREF
  444. _MODULEDEF.fields_by_name['submodules'].message_type = _MODULEDEF
  445. _MODULEDEF.fields_by_name['torchscript_arena'].message_type = _RECORDREF
  446. _MODULEDEF.fields_by_name['caffe2_nets'].message_type = caffe2_dot_proto_dot_caffe2__pb2._NETDEF
  447. _MODULEDEF.fields_by_name['pickle_arena'].message_type = _RECORDREF
  448. _MODULEDEF.fields_by_name['cpp_arena'].message_type = _RECORDREF
  449. _MODULEDEF.fields_by_name['parameters'].message_type = _PARAMETERDEF
  450. _MODULEDEF.fields_by_name['attributes'].message_type = _ATTRIBUTEDEF
  451. _MODULEDEF.fields_by_name['torchscript_debug_arena'].message_type = _RECORDREF
  452. _LIBDEF.fields_by_name['torchscript_arena'].message_type = _RECORDREF
  453. _MODELDEF.fields_by_name['main_module'].message_type = _MODULEDEF
  454. _MODELDEF.fields_by_name['tensors'].message_type = _TENSORDEF
  455. DESCRIPTOR.message_types_by_name['RecordRef'] = _RECORDREF
  456. DESCRIPTOR.message_types_by_name['TensorDef'] = _TENSORDEF
  457. DESCRIPTOR.message_types_by_name['AttributeDef'] = _ATTRIBUTEDEF
  458. DESCRIPTOR.message_types_by_name['ParameterDef'] = _PARAMETERDEF
  459. DESCRIPTOR.message_types_by_name['ModuleDef'] = _MODULEDEF
  460. DESCRIPTOR.message_types_by_name['LibDef'] = _LIBDEF
  461. DESCRIPTOR.message_types_by_name['ModelDef'] = _MODELDEF
  462. DESCRIPTOR.enum_types_by_name['ProtoVersion'] = _PROTOVERSION
  463. _sym_db.RegisterFileDescriptor(DESCRIPTOR)
  464. RecordRef = _reflection.GeneratedProtocolMessageType('RecordRef', (_message.Message,), {
  465. 'DESCRIPTOR' : _RECORDREF,
  466. '__module__' : 'caffe2.proto.torch_pb2'
  467. # @@protoc_insertion_point(class_scope:torch.RecordRef)
  468. })
  469. _sym_db.RegisterMessage(RecordRef)
  470. TensorDef = _reflection.GeneratedProtocolMessageType('TensorDef', (_message.Message,), {
  471. 'DESCRIPTOR' : _TENSORDEF,
  472. '__module__' : 'caffe2.proto.torch_pb2'
  473. # @@protoc_insertion_point(class_scope:torch.TensorDef)
  474. })
  475. _sym_db.RegisterMessage(TensorDef)
  476. AttributeDef = _reflection.GeneratedProtocolMessageType('AttributeDef', (_message.Message,), {
  477. 'DESCRIPTOR' : _ATTRIBUTEDEF,
  478. '__module__' : 'caffe2.proto.torch_pb2'
  479. # @@protoc_insertion_point(class_scope:torch.AttributeDef)
  480. })
  481. _sym_db.RegisterMessage(AttributeDef)
  482. ParameterDef = _reflection.GeneratedProtocolMessageType('ParameterDef', (_message.Message,), {
  483. 'DESCRIPTOR' : _PARAMETERDEF,
  484. '__module__' : 'caffe2.proto.torch_pb2'
  485. # @@protoc_insertion_point(class_scope:torch.ParameterDef)
  486. })
  487. _sym_db.RegisterMessage(ParameterDef)
  488. ModuleDef = _reflection.GeneratedProtocolMessageType('ModuleDef', (_message.Message,), {
  489. 'DESCRIPTOR' : _MODULEDEF,
  490. '__module__' : 'caffe2.proto.torch_pb2'
  491. # @@protoc_insertion_point(class_scope:torch.ModuleDef)
  492. })
  493. _sym_db.RegisterMessage(ModuleDef)
  494. LibDef = _reflection.GeneratedProtocolMessageType('LibDef', (_message.Message,), {
  495. 'DESCRIPTOR' : _LIBDEF,
  496. '__module__' : 'caffe2.proto.torch_pb2'
  497. # @@protoc_insertion_point(class_scope:torch.LibDef)
  498. })
  499. _sym_db.RegisterMessage(LibDef)
  500. ModelDef = _reflection.GeneratedProtocolMessageType('ModelDef', (_message.Message,), {
  501. 'DESCRIPTOR' : _MODELDEF,
  502. '__module__' : 'caffe2.proto.torch_pb2'
  503. # @@protoc_insertion_point(class_scope:torch.ModelDef)
  504. })
  505. _sym_db.RegisterMessage(ModelDef)
  506. # @@protoc_insertion_point(module_scope)