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.

node_def_pb2.py 6.1 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150
  1. # -*- coding: utf-8 -*-
  2. # Generated by the protocol buffer compiler. DO NOT EDIT!
  3. # source: oneflow/customized/utils/node_def.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. from oneflow.customized.utils import attr_value_pb2 as oneflow_dot_customized_dot_utils_dot_attr__value__pb2
  12. DESCRIPTOR = _descriptor.FileDescriptor(
  13. name='oneflow/customized/utils/node_def.proto',
  14. package='oneflow',
  15. syntax='proto3',
  16. serialized_options=None,
  17. create_key=_descriptor._internal_create_key,
  18. serialized_pb=b'\n\'oneflow/customized/utils/node_def.proto\x12\x07oneflow\x1a)oneflow/customized/utils/attr_value.proto\"\xad\x01\n\x07NodeDef\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\n\n\x02op\x18\x02 \x01(\t\x12\r\n\x05input\x18\x03 \x03(\t\x12\x0e\n\x06\x64\x65vice\x18\x04 \x01(\t\x12(\n\x04\x61ttr\x18\x05 \x03(\x0b\x32\x1a.oneflow.NodeDef.AttrEntry\x1a?\n\tAttrEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.oneflow.AttrValue:\x02\x38\x01\x62\x06proto3'
  19. ,
  20. dependencies=[oneflow_dot_customized_dot_utils_dot_attr__value__pb2.DESCRIPTOR,])
  21. _NODEDEF_ATTRENTRY = _descriptor.Descriptor(
  22. name='AttrEntry',
  23. full_name='oneflow.NodeDef.AttrEntry',
  24. filename=None,
  25. file=DESCRIPTOR,
  26. containing_type=None,
  27. create_key=_descriptor._internal_create_key,
  28. fields=[
  29. _descriptor.FieldDescriptor(
  30. name='key', full_name='oneflow.NodeDef.AttrEntry.key', index=0,
  31. number=1, type=9, cpp_type=9, label=1,
  32. has_default_value=False, default_value=b"".decode('utf-8'),
  33. message_type=None, enum_type=None, containing_type=None,
  34. is_extension=False, extension_scope=None,
  35. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  36. _descriptor.FieldDescriptor(
  37. name='value', full_name='oneflow.NodeDef.AttrEntry.value', index=1,
  38. number=2, type=11, cpp_type=10, label=1,
  39. has_default_value=False, default_value=None,
  40. message_type=None, enum_type=None, containing_type=None,
  41. is_extension=False, extension_scope=None,
  42. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  43. ],
  44. extensions=[
  45. ],
  46. nested_types=[],
  47. enum_types=[
  48. ],
  49. serialized_options=b'8\001',
  50. is_extendable=False,
  51. syntax='proto3',
  52. extension_ranges=[],
  53. oneofs=[
  54. ],
  55. serialized_start=206,
  56. serialized_end=269,
  57. )
  58. _NODEDEF = _descriptor.Descriptor(
  59. name='NodeDef',
  60. full_name='oneflow.NodeDef',
  61. filename=None,
  62. file=DESCRIPTOR,
  63. containing_type=None,
  64. create_key=_descriptor._internal_create_key,
  65. fields=[
  66. _descriptor.FieldDescriptor(
  67. name='name', full_name='oneflow.NodeDef.name', index=0,
  68. number=1, type=9, cpp_type=9, label=1,
  69. has_default_value=False, default_value=b"".decode('utf-8'),
  70. message_type=None, enum_type=None, containing_type=None,
  71. is_extension=False, extension_scope=None,
  72. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  73. _descriptor.FieldDescriptor(
  74. name='op', full_name='oneflow.NodeDef.op', index=1,
  75. number=2, type=9, cpp_type=9, label=1,
  76. has_default_value=False, default_value=b"".decode('utf-8'),
  77. message_type=None, enum_type=None, containing_type=None,
  78. is_extension=False, extension_scope=None,
  79. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  80. _descriptor.FieldDescriptor(
  81. name='input', full_name='oneflow.NodeDef.input', index=2,
  82. number=3, type=9, cpp_type=9, label=3,
  83. has_default_value=False, default_value=[],
  84. message_type=None, enum_type=None, containing_type=None,
  85. is_extension=False, extension_scope=None,
  86. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  87. _descriptor.FieldDescriptor(
  88. name='device', full_name='oneflow.NodeDef.device', index=3,
  89. number=4, type=9, cpp_type=9, label=1,
  90. has_default_value=False, default_value=b"".decode('utf-8'),
  91. message_type=None, enum_type=None, containing_type=None,
  92. is_extension=False, extension_scope=None,
  93. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  94. _descriptor.FieldDescriptor(
  95. name='attr', full_name='oneflow.NodeDef.attr', index=4,
  96. number=5, type=11, cpp_type=10, label=3,
  97. has_default_value=False, default_value=[],
  98. message_type=None, enum_type=None, containing_type=None,
  99. is_extension=False, extension_scope=None,
  100. serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),
  101. ],
  102. extensions=[
  103. ],
  104. nested_types=[_NODEDEF_ATTRENTRY, ],
  105. enum_types=[
  106. ],
  107. serialized_options=None,
  108. is_extendable=False,
  109. syntax='proto3',
  110. extension_ranges=[],
  111. oneofs=[
  112. ],
  113. serialized_start=96,
  114. serialized_end=269,
  115. )
  116. _NODEDEF_ATTRENTRY.fields_by_name['value'].message_type = oneflow_dot_customized_dot_utils_dot_attr__value__pb2._ATTRVALUE
  117. _NODEDEF_ATTRENTRY.containing_type = _NODEDEF
  118. _NODEDEF.fields_by_name['attr'].message_type = _NODEDEF_ATTRENTRY
  119. DESCRIPTOR.message_types_by_name['NodeDef'] = _NODEDEF
  120. _sym_db.RegisterFileDescriptor(DESCRIPTOR)
  121. NodeDef = _reflection.GeneratedProtocolMessageType('NodeDef', (_message.Message,), {
  122. 'AttrEntry' : _reflection.GeneratedProtocolMessageType('AttrEntry', (_message.Message,), {
  123. 'DESCRIPTOR' : _NODEDEF_ATTRENTRY,
  124. '__module__' : 'oneflow.customized.utils.node_def_pb2'
  125. # @@protoc_insertion_point(class_scope:oneflow.NodeDef.AttrEntry)
  126. })
  127. ,
  128. 'DESCRIPTOR' : _NODEDEF,
  129. '__module__' : 'oneflow.customized.utils.node_def_pb2'
  130. # @@protoc_insertion_point(class_scope:oneflow.NodeDef)
  131. })
  132. _sym_db.RegisterMessage(NodeDef)
  133. _sym_db.RegisterMessage(NodeDef.AttrEntry)
  134. _NODEDEF_ATTRENTRY._options = None
  135. # @@protoc_insertion_point(module_scope)

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