|
- # Generated by the protocol buffer compiler. DO NOT EDIT!
- # source: oneflow/customized/utils/tensor.proto
-
- import sys
- _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
- from google.protobuf.internal import enum_type_wrapper
- from google.protobuf import descriptor as _descriptor
- from google.protobuf import message as _message
- from google.protobuf import reflection as _reflection
- from google.protobuf import symbol_database as _symbol_database
- from google.protobuf import descriptor_pb2
- # @@protoc_insertion_point(imports)
-
- _sym_db = _symbol_database.Default()
-
-
-
-
- DESCRIPTOR = _descriptor.FileDescriptor(
- name='oneflow/customized/utils/tensor.proto',
- package='oneflow',
- syntax='proto2',
- serialized_pb=_b('\n%oneflow/customized/utils/tensor.proto\x12\x07oneflow\"\x90\x03\n\x0bTensorProto\x12!\n\x05\x64type\x18\x01 \x02(\x0e\x32\x12.oneflow.TDataType\x12/\n\x0ctensor_shape\x18\x02 \x02(\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\x12\x16\n\nuint32_val\x18\x10 \x03(\rB\x02\x10\x01\x12\x16\n\nuint64_val\x18\x11 \x03(\x04\x42\x02\x10\x01\"d\n\x16VariantTensorDataProto\x12\x11\n\ttype_name\x18\x01 \x02(\t\x12\x10\n\x08metadata\x18\x02 \x02(\x0c\x12%\n\x07tensors\x18\x03 \x03(\x0b\x32\x14.oneflow.TensorProto\"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 \x02(\x03\x12\x0c\n\x04name\x18\x02 \x01(\t*\xab\x06\n\tTDataType\x12\x0e\n\nDT_INVALID\x10\x00\x12\x0c\n\x08\x44T_FLOAT\x10\x01\x12\r\n\tDT_DOUBLE\x10\x02\x12\x0c\n\x08\x44T_INT32\x10\x03\x12\x0c\n\x08\x44T_UINT8\x10\x04\x12\x0c\n\x08\x44T_INT16\x10\x05\x12\x0b\n\x07\x44T_INT8\x10\x06\x12\r\n\tDT_STRING\x10\x07\x12\x10\n\x0c\x44T_COMPLEX64\x10\x08\x12\x0c\n\x08\x44T_INT64\x10\t\x12\x0b\n\x07\x44T_BOOL\x10\n\x12\x0c\n\x08\x44T_QINT8\x10\x0b\x12\r\n\tDT_QUINT8\x10\x0c\x12\r\n\tDT_QINT32\x10\r\x12\x0f\n\x0b\x44T_BFLOAT16\x10\x0e\x12\r\n\tDT_QINT16\x10\x0f\x12\x0e\n\nDT_QUINT16\x10\x10\x12\r\n\tDT_UINT16\x10\x11\x12\x11\n\rDT_COMPLEX128\x10\x12\x12\x0b\n\x07\x44T_HALF\x10\x13\x12\x0f\n\x0b\x44T_RESOURCE\x10\x14\x12\x0e\n\nDT_VARIANT\x10\x15\x12\r\n\tDT_UINT32\x10\x16\x12\r\n\tDT_UINT64\x10\x17\x12\x10\n\x0c\x44T_FLOAT_REF\x10\x65\x12\x11\n\rDT_DOUBLE_REF\x10\x66\x12\x10\n\x0c\x44T_INT32_REF\x10g\x12\x10\n\x0c\x44T_UINT8_REF\x10h\x12\x10\n\x0c\x44T_INT16_REF\x10i\x12\x0f\n\x0b\x44T_INT8_REF\x10j\x12\x11\n\rDT_STRING_REF\x10k\x12\x14\n\x10\x44T_COMPLEX64_REF\x10l\x12\x10\n\x0c\x44T_INT64_REF\x10m\x12\x0f\n\x0b\x44T_BOOL_REF\x10n\x12\x10\n\x0c\x44T_QINT8_REF\x10o\x12\x11\n\rDT_QUINT8_REF\x10p\x12\x11\n\rDT_QINT32_REF\x10q\x12\x13\n\x0f\x44T_BFLOAT16_REF\x10r\x12\x11\n\rDT_QINT16_REF\x10s\x12\x12\n\x0e\x44T_QUINT16_REF\x10t\x12\x11\n\rDT_UINT16_REF\x10u\x12\x15\n\x11\x44T_COMPLEX128_REF\x10v\x12\x0f\n\x0b\x44T_HALF_REF\x10w\x12\x13\n\x0f\x44T_RESOURCE_REF\x10x\x12\x12\n\x0e\x44T_VARIANT_REF\x10y\x12\x11\n\rDT_UINT32_REF\x10z\x12\x11\n\rDT_UINT64_REF\x10{')
- )
- _sym_db.RegisterFileDescriptor(DESCRIPTOR)
-
- _TDATATYPE = _descriptor.EnumDescriptor(
- name='TDataType',
- full_name='oneflow.TDataType',
- filename=None,
- file=DESCRIPTOR,
- values=[
- _descriptor.EnumValueDescriptor(
- name='DT_INVALID', index=0, number=0,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_FLOAT', index=1, number=1,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_DOUBLE', index=2, number=2,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_INT32', index=3, number=3,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_UINT8', index=4, number=4,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_INT16', index=5, number=5,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_INT8', index=6, number=6,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_STRING', index=7, number=7,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_COMPLEX64', index=8, number=8,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_INT64', index=9, number=9,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_BOOL', index=10, number=10,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_QINT8', index=11, number=11,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_QUINT8', index=12, number=12,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_QINT32', index=13, number=13,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_BFLOAT16', index=14, number=14,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_QINT16', index=15, number=15,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_QUINT16', index=16, number=16,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_UINT16', index=17, number=17,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_COMPLEX128', index=18, number=18,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_HALF', index=19, number=19,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_RESOURCE', index=20, number=20,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_VARIANT', index=21, number=21,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_UINT32', index=22, number=22,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_UINT64', index=23, number=23,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_FLOAT_REF', index=24, number=101,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_DOUBLE_REF', index=25, number=102,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_INT32_REF', index=26, number=103,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_UINT8_REF', index=27, number=104,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_INT16_REF', index=28, number=105,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_INT8_REF', index=29, number=106,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_STRING_REF', index=30, number=107,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_COMPLEX64_REF', index=31, number=108,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_INT64_REF', index=32, number=109,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_BOOL_REF', index=33, number=110,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_QINT8_REF', index=34, number=111,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_QUINT8_REF', index=35, number=112,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_QINT32_REF', index=36, number=113,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_BFLOAT16_REF', index=37, number=114,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_QINT16_REF', index=38, number=115,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_QUINT16_REF', index=39, number=116,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_UINT16_REF', index=40, number=117,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_COMPLEX128_REF', index=41, number=118,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_HALF_REF', index=42, number=119,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_RESOURCE_REF', index=43, number=120,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_VARIANT_REF', index=44, number=121,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_UINT32_REF', index=45, number=122,
- options=None,
- type=None),
- _descriptor.EnumValueDescriptor(
- name='DT_UINT64_REF', index=46, number=123,
- options=None,
- type=None),
- ],
- containing_type=None,
- options=None,
- serialized_start=677,
- serialized_end=1488,
- )
- _sym_db.RegisterEnumDescriptor(_TDATATYPE)
-
- TDataType = enum_type_wrapper.EnumTypeWrapper(_TDATATYPE)
- DT_INVALID = 0
- DT_FLOAT = 1
- DT_DOUBLE = 2
- DT_INT32 = 3
- DT_UINT8 = 4
- DT_INT16 = 5
- DT_INT8 = 6
- DT_STRING = 7
- DT_COMPLEX64 = 8
- DT_INT64 = 9
- DT_BOOL = 10
- DT_QINT8 = 11
- DT_QUINT8 = 12
- DT_QINT32 = 13
- DT_BFLOAT16 = 14
- DT_QINT16 = 15
- DT_QUINT16 = 16
- DT_UINT16 = 17
- DT_COMPLEX128 = 18
- DT_HALF = 19
- DT_RESOURCE = 20
- DT_VARIANT = 21
- DT_UINT32 = 22
- DT_UINT64 = 23
- DT_FLOAT_REF = 101
- DT_DOUBLE_REF = 102
- DT_INT32_REF = 103
- DT_UINT8_REF = 104
- DT_INT16_REF = 105
- DT_INT8_REF = 106
- DT_STRING_REF = 107
- DT_COMPLEX64_REF = 108
- DT_INT64_REF = 109
- DT_BOOL_REF = 110
- DT_QINT8_REF = 111
- DT_QUINT8_REF = 112
- DT_QINT32_REF = 113
- DT_BFLOAT16_REF = 114
- DT_QINT16_REF = 115
- DT_QUINT16_REF = 116
- DT_UINT16_REF = 117
- DT_COMPLEX128_REF = 118
- DT_HALF_REF = 119
- DT_RESOURCE_REF = 120
- DT_VARIANT_REF = 121
- DT_UINT32_REF = 122
- DT_UINT64_REF = 123
-
-
-
- _TENSORPROTO = _descriptor.Descriptor(
- name='TensorProto',
- full_name='oneflow.TensorProto',
- filename=None,
- file=DESCRIPTOR,
- containing_type=None,
- fields=[
- _descriptor.FieldDescriptor(
- name='dtype', full_name='oneflow.TensorProto.dtype', index=0,
- number=1, type=14, cpp_type=8, label=2,
- has_default_value=False, default_value=0,
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- _descriptor.FieldDescriptor(
- name='tensor_shape', full_name='oneflow.TensorProto.tensor_shape', index=1,
- number=2, type=11, cpp_type=10, label=2,
- has_default_value=False, default_value=None,
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- _descriptor.FieldDescriptor(
- name='version_number', full_name='oneflow.TensorProto.version_number', index=2,
- number=3, type=5, cpp_type=1, label=1,
- has_default_value=False, default_value=0,
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- _descriptor.FieldDescriptor(
- name='tensor_content', full_name='oneflow.TensorProto.tensor_content', index=3,
- number=4, type=12, cpp_type=9, label=1,
- has_default_value=False, default_value=_b(""),
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- _descriptor.FieldDescriptor(
- name='half_val', full_name='oneflow.TensorProto.half_val', index=4,
- number=13, type=5, cpp_type=1, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))),
- _descriptor.FieldDescriptor(
- name='float_val', full_name='oneflow.TensorProto.float_val', index=5,
- number=5, type=2, cpp_type=6, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))),
- _descriptor.FieldDescriptor(
- name='double_val', full_name='oneflow.TensorProto.double_val', index=6,
- number=6, type=1, cpp_type=5, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))),
- _descriptor.FieldDescriptor(
- name='int_val', full_name='oneflow.TensorProto.int_val', index=7,
- number=7, type=5, cpp_type=1, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))),
- _descriptor.FieldDescriptor(
- name='string_val', full_name='oneflow.TensorProto.string_val', index=8,
- number=8, type=12, cpp_type=9, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- _descriptor.FieldDescriptor(
- name='scomplex_val', full_name='oneflow.TensorProto.scomplex_val', index=9,
- number=9, type=2, cpp_type=6, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))),
- _descriptor.FieldDescriptor(
- name='int64_val', full_name='oneflow.TensorProto.int64_val', index=10,
- number=10, type=3, cpp_type=2, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))),
- _descriptor.FieldDescriptor(
- name='bool_val', full_name='oneflow.TensorProto.bool_val', index=11,
- number=11, type=8, cpp_type=7, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))),
- _descriptor.FieldDescriptor(
- name='dcomplex_val', full_name='oneflow.TensorProto.dcomplex_val', index=12,
- number=12, type=1, cpp_type=5, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))),
- _descriptor.FieldDescriptor(
- name='uint32_val', full_name='oneflow.TensorProto.uint32_val', index=13,
- number=16, type=13, cpp_type=3, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))),
- _descriptor.FieldDescriptor(
- name='uint64_val', full_name='oneflow.TensorProto.uint64_val', index=14,
- number=17, type=4, cpp_type=4, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))),
- ],
- extensions=[
- ],
- nested_types=[],
- enum_types=[
- ],
- options=None,
- is_extendable=False,
- syntax='proto2',
- extension_ranges=[],
- oneofs=[
- ],
- serialized_start=51,
- serialized_end=451,
- )
-
-
- _VARIANTTENSORDATAPROTO = _descriptor.Descriptor(
- name='VariantTensorDataProto',
- full_name='oneflow.VariantTensorDataProto',
- filename=None,
- file=DESCRIPTOR,
- containing_type=None,
- fields=[
- _descriptor.FieldDescriptor(
- name='type_name', full_name='oneflow.VariantTensorDataProto.type_name', index=0,
- number=1, type=9, cpp_type=9, label=2,
- has_default_value=False, default_value=_b("").decode('utf-8'),
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- _descriptor.FieldDescriptor(
- name='metadata', full_name='oneflow.VariantTensorDataProto.metadata', index=1,
- number=2, type=12, cpp_type=9, label=2,
- has_default_value=False, default_value=_b(""),
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- _descriptor.FieldDescriptor(
- name='tensors', full_name='oneflow.VariantTensorDataProto.tensors', index=2,
- number=3, type=11, cpp_type=10, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- ],
- extensions=[
- ],
- nested_types=[],
- enum_types=[
- ],
- options=None,
- is_extendable=False,
- syntax='proto2',
- extension_ranges=[],
- oneofs=[
- ],
- serialized_start=453,
- serialized_end=553,
- )
-
-
- _TENSORSHAPEPROTO_DIM = _descriptor.Descriptor(
- name='Dim',
- full_name='oneflow.TensorShapeProto.Dim',
- filename=None,
- file=DESCRIPTOR,
- containing_type=None,
- fields=[
- _descriptor.FieldDescriptor(
- name='size', full_name='oneflow.TensorShapeProto.Dim.size', index=0,
- number=1, type=3, cpp_type=2, label=2,
- has_default_value=False, default_value=0,
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- _descriptor.FieldDescriptor(
- name='name', full_name='oneflow.TensorShapeProto.Dim.name', index=1,
- number=2, type=9, cpp_type=9, label=1,
- has_default_value=False, default_value=_b("").decode('utf-8'),
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- ],
- extensions=[
- ],
- nested_types=[],
- enum_types=[
- ],
- options=None,
- is_extendable=False,
- syntax='proto2',
- extension_ranges=[],
- oneofs=[
- ],
- serialized_start=641,
- serialized_end=674,
- )
-
- _TENSORSHAPEPROTO = _descriptor.Descriptor(
- name='TensorShapeProto',
- full_name='oneflow.TensorShapeProto',
- filename=None,
- file=DESCRIPTOR,
- containing_type=None,
- fields=[
- _descriptor.FieldDescriptor(
- name='dim', full_name='oneflow.TensorShapeProto.dim', index=0,
- number=2, type=11, cpp_type=10, label=3,
- has_default_value=False, default_value=[],
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- _descriptor.FieldDescriptor(
- name='unknown_rank', full_name='oneflow.TensorShapeProto.unknown_rank', index=1,
- number=3, type=8, cpp_type=7, label=1,
- has_default_value=False, default_value=False,
- message_type=None, enum_type=None, containing_type=None,
- is_extension=False, extension_scope=None,
- options=None),
- ],
- extensions=[
- ],
- nested_types=[_TENSORSHAPEPROTO_DIM, ],
- enum_types=[
- ],
- options=None,
- is_extendable=False,
- syntax='proto2',
- extension_ranges=[],
- oneofs=[
- ],
- serialized_start=555,
- serialized_end=674,
- )
-
- _TENSORPROTO.fields_by_name['dtype'].enum_type = _TDATATYPE
- _TENSORPROTO.fields_by_name['tensor_shape'].message_type = _TENSORSHAPEPROTO
- _VARIANTTENSORDATAPROTO.fields_by_name['tensors'].message_type = _TENSORPROTO
- _TENSORSHAPEPROTO_DIM.containing_type = _TENSORSHAPEPROTO
- _TENSORSHAPEPROTO.fields_by_name['dim'].message_type = _TENSORSHAPEPROTO_DIM
- DESCRIPTOR.message_types_by_name['TensorProto'] = _TENSORPROTO
- DESCRIPTOR.message_types_by_name['VariantTensorDataProto'] = _VARIANTTENSORDATAPROTO
- DESCRIPTOR.message_types_by_name['TensorShapeProto'] = _TENSORSHAPEPROTO
- DESCRIPTOR.enum_types_by_name['TDataType'] = _TDATATYPE
-
- TensorProto = _reflection.GeneratedProtocolMessageType('TensorProto', (_message.Message,), dict(
- DESCRIPTOR = _TENSORPROTO,
- __module__ = 'oneflow.customized.utils.tensor_pb2'
- # @@protoc_insertion_point(class_scope:oneflow.TensorProto)
- ))
- _sym_db.RegisterMessage(TensorProto)
-
- VariantTensorDataProto = _reflection.GeneratedProtocolMessageType('VariantTensorDataProto', (_message.Message,), dict(
- DESCRIPTOR = _VARIANTTENSORDATAPROTO,
- __module__ = 'oneflow.customized.utils.tensor_pb2'
- # @@protoc_insertion_point(class_scope:oneflow.VariantTensorDataProto)
- ))
- _sym_db.RegisterMessage(VariantTensorDataProto)
-
- TensorShapeProto = _reflection.GeneratedProtocolMessageType('TensorShapeProto', (_message.Message,), dict(
-
- Dim = _reflection.GeneratedProtocolMessageType('Dim', (_message.Message,), dict(
- DESCRIPTOR = _TENSORSHAPEPROTO_DIM,
- __module__ = 'oneflow.customized.utils.tensor_pb2'
- # @@protoc_insertion_point(class_scope:oneflow.TensorShapeProto.Dim)
- ))
- ,
- DESCRIPTOR = _TENSORSHAPEPROTO,
- __module__ = 'oneflow.customized.utils.tensor_pb2'
- # @@protoc_insertion_point(class_scope:oneflow.TensorShapeProto)
- ))
- _sym_db.RegisterMessage(TensorShapeProto)
- _sym_db.RegisterMessage(TensorShapeProto.Dim)
-
-
- _TENSORPROTO.fields_by_name['half_val'].has_options = True
- _TENSORPROTO.fields_by_name['half_val']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))
- _TENSORPROTO.fields_by_name['float_val'].has_options = True
- _TENSORPROTO.fields_by_name['float_val']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))
- _TENSORPROTO.fields_by_name['double_val'].has_options = True
- _TENSORPROTO.fields_by_name['double_val']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))
- _TENSORPROTO.fields_by_name['int_val'].has_options = True
- _TENSORPROTO.fields_by_name['int_val']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))
- _TENSORPROTO.fields_by_name['scomplex_val'].has_options = True
- _TENSORPROTO.fields_by_name['scomplex_val']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))
- _TENSORPROTO.fields_by_name['int64_val'].has_options = True
- _TENSORPROTO.fields_by_name['int64_val']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))
- _TENSORPROTO.fields_by_name['bool_val'].has_options = True
- _TENSORPROTO.fields_by_name['bool_val']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))
- _TENSORPROTO.fields_by_name['dcomplex_val'].has_options = True
- _TENSORPROTO.fields_by_name['dcomplex_val']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))
- _TENSORPROTO.fields_by_name['uint32_val'].has_options = True
- _TENSORPROTO.fields_by_name['uint32_val']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))
- _TENSORPROTO.fields_by_name['uint64_val'].has_options = True
- _TENSORPROTO.fields_by_name['uint64_val']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\020\001'))
- # @@protoc_insertion_point(module_scope)
|