You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

tensor_pb2.py 12 kB

5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258
  1. # -*- coding: utf-8 -*-
  2. # Generated by the protocol buffer compiler. DO NOT EDIT!
  3. # source: oneflow/customized/utils/tensor.proto
  4. """Generated protocol buffer code."""
  5. from google.protobuf import descriptor as _descriptor
  6. from google.protobuf import message as _message
  7. from google.protobuf import reflection as _reflection
  8. from google.protobuf import symbol_database as _symbol_database
  9. # @@protoc_insertion_point(imports)
  10. _sym_db = _symbol_database.Default()
  11. DESCRIPTOR = _descriptor.FileDescriptor(
  12. name='oneflow/customized/utils/tensor.proto',
  13. package='oneflow',
  14. syntax='proto3',
  15. serialized_options=None,
  16. create_key=_descriptor._internal_create_key,
  17. serialized_pb=b'\n%oneflow/customized/utils/tensor.proto\x12\x07oneflow\"w\n\x10TensorShapeProto\x12*\n\x03\x64im\x18\x02 \x03(\x0b\x32\x1d.oneflow.TensorShapeProto.Dim\x12\x14\n\x0cunknown_rank\x18\x03 \x01(\x08\x1a!\n\x03\x44im\x12\x0c\n\x04size\x18\x01 \x01(\x03\x12\x0c\n\x04name\x18\x02 \x01(\t\"\xcc\x02\n\x0bTensorProto\x12\r\n\x05\x64type\x18\x01 \x01(\t\x12/\n\x0ctensor_shape\x18\x02 \x01(\x0b\x32\x19.oneflow.TensorShapeProto\x12\x16\n\x0eversion_number\x18\x03 \x01(\x05\x12\x16\n\x0etensor_content\x18\x04 \x01(\x0c\x12\x14\n\x08half_val\x18\r \x03(\x05\x42\x02\x10\x01\x12\x15\n\tfloat_val\x18\x05 \x03(\x02\x42\x02\x10\x01\x12\x16\n\ndouble_val\x18\x06 \x03(\x01\x42\x02\x10\x01\x12\x13\n\x07int_val\x18\x07 \x03(\x05\x42\x02\x10\x01\x12\x12\n\nstring_val\x18\x08 \x03(\x0c\x12\x18\n\x0cscomplex_val\x18\t \x03(\x02\x42\x02\x10\x01\x12\x15\n\tint64_val\x18\n \x03(\x03\x42\x02\x10\x01\x12\x14\n\x08\x62ool_val\x18\x0b \x03(\x08\x42\x02\x10\x01\x12\x18\n\x0c\x64\x63omplex_val\x18\x0c \x03(\x01\x42\x02\x10\x01\x62\x06proto3'
  18. )
  19. _TENSORSHAPEPROTO_DIM = _descriptor.Descriptor(
  20. name='Dim',
  21. full_name='oneflow.TensorShapeProto.Dim',
  22. filename=None,
  23. file=DESCRIPTOR,
  24. containing_type=None,
  25. create_key=_descriptor._internal_create_key,
  26. fields=[
  27. _descriptor.FieldDescriptor(
  28. name='size', full_name='oneflow.TensorShapeProto.Dim.size', index=0,
  29. number=1, type=3, cpp_type=2, label=1,
  30. has_default_value=False, default_value=0,
  31. message_type=None, enum_type=None, containing_type=None,
  32. is_extension=False, extension_scope=None,
  33. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  34. _descriptor.FieldDescriptor(
  35. name='name', full_name='oneflow.TensorShapeProto.Dim.name', index=1,
  36. number=2, type=9, cpp_type=9, label=1,
  37. has_default_value=False, default_value=b"".decode('utf-8'),
  38. message_type=None, enum_type=None, containing_type=None,
  39. is_extension=False, extension_scope=None,
  40. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  41. ],
  42. extensions=[
  43. ],
  44. nested_types=[],
  45. enum_types=[
  46. ],
  47. serialized_options=None,
  48. is_extendable=False,
  49. syntax='proto3',
  50. extension_ranges=[],
  51. oneofs=[
  52. ],
  53. serialized_start=136,
  54. serialized_end=169,
  55. )
  56. _TENSORSHAPEPROTO = _descriptor.Descriptor(
  57. name='TensorShapeProto',
  58. full_name='oneflow.TensorShapeProto',
  59. filename=None,
  60. file=DESCRIPTOR,
  61. containing_type=None,
  62. create_key=_descriptor._internal_create_key,
  63. fields=[
  64. _descriptor.FieldDescriptor(
  65. name='dim', full_name='oneflow.TensorShapeProto.dim', index=0,
  66. number=2, type=11, cpp_type=10, label=3,
  67. has_default_value=False, default_value=[],
  68. message_type=None, enum_type=None, containing_type=None,
  69. is_extension=False, extension_scope=None,
  70. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  71. _descriptor.FieldDescriptor(
  72. name='unknown_rank', full_name='oneflow.TensorShapeProto.unknown_rank', index=1,
  73. number=3, type=8, cpp_type=7, label=1,
  74. has_default_value=False, default_value=False,
  75. message_type=None, enum_type=None, containing_type=None,
  76. is_extension=False, extension_scope=None,
  77. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  78. ],
  79. extensions=[
  80. ],
  81. nested_types=[_TENSORSHAPEPROTO_DIM, ],
  82. enum_types=[
  83. ],
  84. serialized_options=None,
  85. is_extendable=False,
  86. syntax='proto3',
  87. extension_ranges=[],
  88. oneofs=[
  89. ],
  90. serialized_start=50,
  91. serialized_end=169,
  92. )
  93. _TENSORPROTO = _descriptor.Descriptor(
  94. name='TensorProto',
  95. full_name='oneflow.TensorProto',
  96. filename=None,
  97. file=DESCRIPTOR,
  98. containing_type=None,
  99. create_key=_descriptor._internal_create_key,
  100. fields=[
  101. _descriptor.FieldDescriptor(
  102. name='dtype', full_name='oneflow.TensorProto.dtype', index=0,
  103. number=1, type=9, cpp_type=9, label=1,
  104. has_default_value=False, default_value=b"".decode('utf-8'),
  105. message_type=None, enum_type=None, containing_type=None,
  106. is_extension=False, extension_scope=None,
  107. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  108. _descriptor.FieldDescriptor(
  109. name='tensor_shape', full_name='oneflow.TensorProto.tensor_shape', index=1,
  110. number=2, type=11, cpp_type=10, label=1,
  111. has_default_value=False, default_value=None,
  112. message_type=None, enum_type=None, containing_type=None,
  113. is_extension=False, extension_scope=None,
  114. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  115. _descriptor.FieldDescriptor(
  116. name='version_number', full_name='oneflow.TensorProto.version_number', index=2,
  117. number=3, type=5, cpp_type=1, label=1,
  118. has_default_value=False, default_value=0,
  119. message_type=None, enum_type=None, containing_type=None,
  120. is_extension=False, extension_scope=None,
  121. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  122. _descriptor.FieldDescriptor(
  123. name='tensor_content', full_name='oneflow.TensorProto.tensor_content', index=3,
  124. number=4, type=12, cpp_type=9, label=1,
  125. has_default_value=False, default_value=b"",
  126. message_type=None, enum_type=None, containing_type=None,
  127. is_extension=False, extension_scope=None,
  128. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  129. _descriptor.FieldDescriptor(
  130. name='half_val', full_name='oneflow.TensorProto.half_val', index=4,
  131. number=13, type=5, cpp_type=1, label=3,
  132. has_default_value=False, default_value=[],
  133. message_type=None, enum_type=None, containing_type=None,
  134. is_extension=False, extension_scope=None,
  135. serialized_options=b'\020\001', file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  136. _descriptor.FieldDescriptor(
  137. name='float_val', full_name='oneflow.TensorProto.float_val', index=5,
  138. number=5, type=2, cpp_type=6, label=3,
  139. has_default_value=False, default_value=[],
  140. message_type=None, enum_type=None, containing_type=None,
  141. is_extension=False, extension_scope=None,
  142. serialized_options=b'\020\001', file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  143. _descriptor.FieldDescriptor(
  144. name='double_val', full_name='oneflow.TensorProto.double_val', index=6,
  145. number=6, type=1, cpp_type=5, label=3,
  146. has_default_value=False, default_value=[],
  147. message_type=None, enum_type=None, containing_type=None,
  148. is_extension=False, extension_scope=None,
  149. serialized_options=b'\020\001', file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  150. _descriptor.FieldDescriptor(
  151. name='int_val', full_name='oneflow.TensorProto.int_val', index=7,
  152. number=7, type=5, cpp_type=1, label=3,
  153. has_default_value=False, default_value=[],
  154. message_type=None, enum_type=None, containing_type=None,
  155. is_extension=False, extension_scope=None,
  156. serialized_options=b'\020\001', file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  157. _descriptor.FieldDescriptor(
  158. name='string_val', full_name='oneflow.TensorProto.string_val', index=8,
  159. number=8, type=12, cpp_type=9, label=3,
  160. has_default_value=False, default_value=[],
  161. message_type=None, enum_type=None, containing_type=None,
  162. is_extension=False, extension_scope=None,
  163. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  164. _descriptor.FieldDescriptor(
  165. name='scomplex_val', full_name='oneflow.TensorProto.scomplex_val', index=9,
  166. number=9, type=2, cpp_type=6, label=3,
  167. has_default_value=False, default_value=[],
  168. message_type=None, enum_type=None, containing_type=None,
  169. is_extension=False, extension_scope=None,
  170. serialized_options=b'\020\001', file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  171. _descriptor.FieldDescriptor(
  172. name='int64_val', full_name='oneflow.TensorProto.int64_val', index=10,
  173. number=10, type=3, cpp_type=2, label=3,
  174. has_default_value=False, default_value=[],
  175. message_type=None, enum_type=None, containing_type=None,
  176. is_extension=False, extension_scope=None,
  177. serialized_options=b'\020\001', file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  178. _descriptor.FieldDescriptor(
  179. name='bool_val', full_name='oneflow.TensorProto.bool_val', index=11,
  180. number=11, type=8, cpp_type=7, label=3,
  181. has_default_value=False, default_value=[],
  182. message_type=None, enum_type=None, containing_type=None,
  183. is_extension=False, extension_scope=None,
  184. serialized_options=b'\020\001', file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  185. _descriptor.FieldDescriptor(
  186. name='dcomplex_val', full_name='oneflow.TensorProto.dcomplex_val', index=12,
  187. number=12, type=1, cpp_type=5, label=3,
  188. has_default_value=False, default_value=[],
  189. message_type=None, enum_type=None, containing_type=None,
  190. is_extension=False, extension_scope=None,
  191. serialized_options=b'\020\001', file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  192. ],
  193. extensions=[
  194. ],
  195. nested_types=[],
  196. enum_types=[
  197. ],
  198. serialized_options=None,
  199. is_extendable=False,
  200. syntax='proto3',
  201. extension_ranges=[],
  202. oneofs=[
  203. ],
  204. serialized_start=172,
  205. serialized_end=504,
  206. )
  207. _TENSORSHAPEPROTO_DIM.containing_type = _TENSORSHAPEPROTO
  208. _TENSORSHAPEPROTO.fields_by_name['dim'].message_type = _TENSORSHAPEPROTO_DIM
  209. _TENSORPROTO.fields_by_name['tensor_shape'].message_type = _TENSORSHAPEPROTO
  210. DESCRIPTOR.message_types_by_name['TensorShapeProto'] = _TENSORSHAPEPROTO
  211. DESCRIPTOR.message_types_by_name['TensorProto'] = _TENSORPROTO
  212. _sym_db.RegisterFileDescriptor(DESCRIPTOR)
  213. TensorShapeProto = _reflection.GeneratedProtocolMessageType('TensorShapeProto', (_message.Message,), {
  214. 'Dim' : _reflection.GeneratedProtocolMessageType('Dim', (_message.Message,), {
  215. 'DESCRIPTOR' : _TENSORSHAPEPROTO_DIM,
  216. '__module__' : 'oneflow.customized.utils.tensor_pb2'
  217. # @@protoc_insertion_point(class_scope:oneflow.TensorShapeProto.Dim)
  218. })
  219. ,
  220. 'DESCRIPTOR' : _TENSORSHAPEPROTO,
  221. '__module__' : 'oneflow.customized.utils.tensor_pb2'
  222. # @@protoc_insertion_point(class_scope:oneflow.TensorShapeProto)
  223. })
  224. _sym_db.RegisterMessage(TensorShapeProto)
  225. _sym_db.RegisterMessage(TensorShapeProto.Dim)
  226. TensorProto = _reflection.GeneratedProtocolMessageType('TensorProto', (_message.Message,), {
  227. 'DESCRIPTOR' : _TENSORPROTO,
  228. '__module__' : 'oneflow.customized.utils.tensor_pb2'
  229. # @@protoc_insertion_point(class_scope:oneflow.TensorProto)
  230. })
  231. _sym_db.RegisterMessage(TensorProto)
  232. _TENSORPROTO.fields_by_name['half_val']._options = None
  233. _TENSORPROTO.fields_by_name['float_val']._options = None
  234. _TENSORPROTO.fields_by_name['double_val']._options = None
  235. _TENSORPROTO.fields_by_name['int_val']._options = None
  236. _TENSORPROTO.fields_by_name['scomplex_val']._options = None
  237. _TENSORPROTO.fields_by_name['int64_val']._options = None
  238. _TENSORPROTO.fields_by_name['bool_val']._options = None
  239. _TENSORPROTO.fields_by_name['dcomplex_val']._options = None
  240. # @@protoc_insertion_point(module_scope)

一站式算法开发平台、高性能分布式深度学习框架、先进算法模型库、视觉模型炼知平台、数据可视化分析平台等一系列平台及工具,在模型高效分布式训练、数据处理和可视分析、模型炼知和轻量化等技术上形成独特优势,目前已在产学研等各领域近千家单位及个人提供AI应用赋能