tensorbored 2.21.0rc1769983804__py3-none-any.whl

tensorbored/compat/proto/tensor_description_pb2.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/tensor_description.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from tensorbored.compat.proto import allocation_description_pb2 as tensorbored_dot_compat_dot_proto_dot_allocation__description__pb2
+from tensorbored.compat.proto import tensor_shape_pb2 as tensorbored_dot_compat_dot_proto_dot_tensor__shape__pb2
+from tensorbored.compat.proto import types_pb2 as tensorbored_dot_compat_dot_proto_dot_types__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1tensorbored/compat/proto/tensor_description.proto\x12\x0btensorbored\x1a\x35tensorbored/compat/proto/allocation_description.proto\x1a+tensorbored/compat/proto/tensor_shape.proto\x1a$tensorbored/compat/proto/types.proto\"\xab\x01\n\x11TensorDescription\x12$\n\x05\x64type\x18\x01 \x01(\x0e\x32\x15.tensorbored.DataType\x12,\n\x05shape\x18\x02 \x01(\x0b\x32\x1d.tensorbored.TensorShapeProto\x12\x42\n\x16\x61llocation_description\x18\x04 \x01(\x0b\x32\".tensorbored.AllocationDescriptionB\x93\x01\n\x18org.tensorflow.frameworkB\x17TensorDescriptionProtosP\x01ZYgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_description_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_TENSORDESCRIPTION = DESCRIPTOR.message_types_by_name['TensorDescription']
+TensorDescription = _reflection.GeneratedProtocolMessageType('TensorDescription', (_message.Message,), {
+  'DESCRIPTOR' : _TENSORDESCRIPTION,
+  '__module__' : 'tensorbored.compat.proto.tensor_description_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.TensorDescription)
+  })
+_sym_db.RegisterMessage(TensorDescription)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\027TensorDescriptionProtosP\001ZYgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_description_go_proto\370\001\001'
+  _TENSORDESCRIPTION._serialized_start=205
+  _TENSORDESCRIPTION._serialized_end=376
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/tensor_pb2.py

@@ -0,0 +1,68 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/tensor.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from tensorbored.compat.proto import resource_handle_pb2 as tensorbored_dot_compat_dot_proto_dot_resource__handle__pb2
+from tensorbored.compat.proto import tensor_shape_pb2 as tensorbored_dot_compat_dot_proto_dot_tensor__shape__pb2
+from tensorbored.compat.proto import types_pb2 as tensorbored_dot_compat_dot_proto_dot_types__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%tensorbored/compat/proto/tensor.proto\x12\x0btensorbored\x1a.tensorbored/compat/proto/resource_handle.proto\x1a+tensorbored/compat/proto/tensor_shape.proto\x1a$tensorbored/compat/proto/types.proto\"\xa4\x04\n\x0bTensorProto\x12$\n\x05\x64type\x18\x01 \x01(\x0e\x32\x15.tensorbored.DataType\x12\x33\n\x0ctensor_shape\x18\x02 \x01(\x0b\x32\x1d.tensorbored.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=\n\x13resource_handle_val\x18\x0e \x03(\x0b\x32 .tensorbored.ResourceHandleProto\x12\x38\n\x0bvariant_val\x18\x0f \x03(\x0b\x32#.tensorbored.VariantTensorDataProto\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\x12\x12\n\nfloat8_val\x18\x12 \x01(\x0c\"h\n\x16VariantTensorDataProto\x12\x11\n\ttype_name\x18\x01 \x01(\t\x12\x10\n\x08metadata\x18\x02 \x01(\x0c\x12)\n\x07tensors\x18\x03 \x03(\x0b\x32\x18.tensorbored.TensorProtoB|\n\x18org.tensorflow.frameworkB\x0cTensorProtosP\x01ZMgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_TENSORPROTO = DESCRIPTOR.message_types_by_name['TensorProto']
+_VARIANTTENSORDATAPROTO = DESCRIPTOR.message_types_by_name['VariantTensorDataProto']
+TensorProto = _reflection.GeneratedProtocolMessageType('TensorProto', (_message.Message,), {
+  'DESCRIPTOR' : _TENSORPROTO,
+  '__module__' : 'tensorbored.compat.proto.tensor_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.TensorProto)
+  })
+_sym_db.RegisterMessage(TensorProto)
+
+VariantTensorDataProto = _reflection.GeneratedProtocolMessageType('VariantTensorDataProto', (_message.Message,), {
+  'DESCRIPTOR' : _VARIANTTENSORDATAPROTO,
+  '__module__' : 'tensorbored.compat.proto.tensor_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.VariantTensorDataProto)
+  })
+_sym_db.RegisterMessage(VariantTensorDataProto)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\014TensorProtosP\001ZMgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_go_proto\370\001\001'
+  _TENSORPROTO.fields_by_name['half_val']._options = None
+  _TENSORPROTO.fields_by_name['half_val']._serialized_options = b'\020\001'
+  _TENSORPROTO.fields_by_name['float_val']._options = None
+  _TENSORPROTO.fields_by_name['float_val']._serialized_options = b'\020\001'
+  _TENSORPROTO.fields_by_name['double_val']._options = None
+  _TENSORPROTO.fields_by_name['double_val']._serialized_options = b'\020\001'
+  _TENSORPROTO.fields_by_name['int_val']._options = None
+  _TENSORPROTO.fields_by_name['int_val']._serialized_options = b'\020\001'
+  _TENSORPROTO.fields_by_name['scomplex_val']._options = None
+  _TENSORPROTO.fields_by_name['scomplex_val']._serialized_options = b'\020\001'
+  _TENSORPROTO.fields_by_name['int64_val']._options = None
+  _TENSORPROTO.fields_by_name['int64_val']._serialized_options = b'\020\001'
+  _TENSORPROTO.fields_by_name['bool_val']._options = None
+  _TENSORPROTO.fields_by_name['bool_val']._serialized_options = b'\020\001'
+  _TENSORPROTO.fields_by_name['dcomplex_val']._options = None
+  _TENSORPROTO.fields_by_name['dcomplex_val']._serialized_options = b'\020\001'
+  _TENSORPROTO.fields_by_name['uint32_val']._options = None
+  _TENSORPROTO.fields_by_name['uint32_val']._serialized_options = b'\020\001'
+  _TENSORPROTO.fields_by_name['uint64_val']._options = None
+  _TENSORPROTO.fields_by_name['uint64_val']._serialized_options = b'\020\001'
+  _TENSORPROTO._serialized_start=186
+  _TENSORPROTO._serialized_end=734
+  _VARIANTTENSORDATAPROTO._serialized_start=736
+  _VARIANTTENSORDATAPROTO._serialized_end=840
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/tensor_shape_pb2.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/tensor_shape.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+tensorbored/compat/proto/tensor_shape.proto\x12\x0btensorbored\"{\n\x10TensorShapeProto\x12.\n\x03\x64im\x18\x02 \x03(\x0b\x32!.tensorbored.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(\tB\x87\x01\n\x18org.tensorflow.frameworkB\x11TensorShapeProtosP\x01ZSgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_shape_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_TENSORSHAPEPROTO = DESCRIPTOR.message_types_by_name['TensorShapeProto']
+_TENSORSHAPEPROTO_DIM = _TENSORSHAPEPROTO.nested_types_by_name['Dim']
+TensorShapeProto = _reflection.GeneratedProtocolMessageType('TensorShapeProto', (_message.Message,), {
+
+  'Dim' : _reflection.GeneratedProtocolMessageType('Dim', (_message.Message,), {
+    'DESCRIPTOR' : _TENSORSHAPEPROTO_DIM,
+    '__module__' : 'tensorbored.compat.proto.tensor_shape_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.TensorShapeProto.Dim)
+    })
+  ,
+  'DESCRIPTOR' : _TENSORSHAPEPROTO,
+  '__module__' : 'tensorbored.compat.proto.tensor_shape_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.TensorShapeProto)
+  })
+_sym_db.RegisterMessage(TensorShapeProto)
+_sym_db.RegisterMessage(TensorShapeProto.Dim)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\021TensorShapeProtosP\001ZSgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_shape_go_proto\370\001\001'
+  _TENSORSHAPEPROTO._serialized_start=60
+  _TENSORSHAPEPROTO._serialized_end=183
+  _TENSORSHAPEPROTO_DIM._serialized_start=150
+  _TENSORSHAPEPROTO_DIM._serialized_end=183
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/tfprof_log_pb2.py

@@ -0,0 +1,307 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/tfprof_log.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from tensorbored.compat.proto import attr_value_pb2 as tensorbored_dot_compat_dot_proto_dot_attr__value__pb2
+from tensorbored.compat.proto import step_stats_pb2 as tensorbored_dot_compat_dot_proto_dot_step__stats__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)tensorbored/compat/proto/tfprof_log.proto\x12\x0btensorbored\x1a)tensorbored/compat/proto/attr_value.proto\x1a)tensorbored/compat/proto/step_stats.proto\"\xc7\x01\n\x07\x43odeDef\x12*\n\x06traces\x18\x01 \x03(\x0b\x32\x1a.tensorbored.CodeDef.Trace\x1a\x8f\x01\n\x05Trace\x12\x0f\n\x07\x66ile_id\x18\x06 \x01(\x03\x12\x0e\n\x06lineno\x18\x02 \x01(\x05\x12\x13\n\x0b\x66unction_id\x18\x07 \x01(\x03\x12\x0f\n\x07line_id\x18\x08 \x01(\x03\x12\x17\n\x0f\x66unc_start_line\x18\x05 \x01(\x05J\x04\x08\x01\x10\x02J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05R\x04\x66ileR\x08\x66unctionR\x04line\"d\n\nOpLogEntry\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\tfloat_ops\x18\x02 \x01(\x03\x12\r\n\x05types\x18\x03 \x03(\t\x12&\n\x08\x63ode_def\x18\x04 \x01(\x0b\x32\x14.tensorbored.CodeDef\"\xac\x01\n\nOpLogProto\x12,\n\x0blog_entries\x18\x01 \x03(\x0b\x32\x17.tensorbored.OpLogEntry\x12=\n\x0cid_to_string\x18\x02 \x03(\x0b\x32\'.tensorbored.OpLogProto.IdToStringEntry\x1a\x31\n\x0fIdToStringEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xc2\x02\n\x0cProfileProto\x12\x33\n\x05nodes\x18\x01 \x03(\x0b\x32$.tensorbored.ProfileProto.NodesEntry\x12\x11\n\thas_trace\x18\x02 \x01(\x08\x12\x1f\n\x17miss_accelerator_stream\x18\x05 \x01(\x08\x12\r\n\x05steps\x18\x03 \x03(\x03\x12?\n\x0cid_to_string\x18\x04 \x03(\x0b\x32).tensorbored.ProfileProto.IdToStringEntry\x1a\x46\n\nNodesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\'\n\x05value\x18\x02 \x01(\x0b\x32\x18.tensorbored.ProfileNode:\x02\x38\x01\x1a\x31\n\x0fIdToStringEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x92\x08\n\x0bProfileNode\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\n\n\x02op\x18\t \x01(\t\x12\n\n\x02id\x18\r \x01(\x03\x12\x34\n\x06inputs\x18\x02 \x03(\x0b\x32$.tensorbored.ProfileNode.InputsEntry\x12?\n\x0cinput_shapes\x18\x10 \x03(\x0b\x32).tensorbored.ProfileNode.InputShapesEntry\x12\x36\n\x07outputs\x18\x03 \x03(\x0b\x32%.tensorbored.ProfileNode.OutputsEntry\x12\x41\n\routput_shapes\x18\x0f \x03(\x0b\x32*.tensorbored.ProfileNode.OutputShapesEntry\x12\x46\n\x10src_output_index\x18\x0e \x03(\x0b\x32,.tensorbored.ProfileNode.SrcOutputIndexEntry\x12\r\n\x05shape\x18\x04 \x03(\x03\x12\x10\n\x08op_types\x18\x05 \x03(\t\x12\x18\n\x10\x63\x61nonical_device\x18\x06 \x01(\t\x12\x13\n\x0bhost_device\x18\x07 \x01(\t\x12\x11\n\tfloat_ops\x18\x08 \x01(\x03\x12#\n\x05trace\x18\n \x01(\x0b\x32\x14.tensorbored.CodeDef\x12\x32\n\x05\x61ttrs\x18\x0b \x03(\x0b\x32#.tensorbored.ProfileNode.AttrsEntry\x12\x32\n\x05\x65xecs\x18\x0c \x03(\x0b\x32#.tensorbored.ProfileNode.ExecsEntry\x1a-\n\x0bInputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\r\n\x05value\x18\x02 \x01(\x03:\x02\x38\x01\x1a\x46\n\x10InputShapesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.tensorbored.Tuple:\x02\x38\x01\x1a.\n\x0cOutputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\r\n\x05value\x18\x02 \x01(\x03:\x02\x38\x01\x1aG\n\x11OutputShapesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.tensorbored.Tuple:\x02\x38\x01\x1a\x35\n\x13SrcOutputIndexEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\x1a\x44\n\nAttrsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.tensorbored.AttrValue:\x02\x38\x01\x1a\x46\n\nExecsEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\'\n\x05value\x18\x02 \x01(\x0b\x32\x18.tensorbored.ExecProfile:\x02\x38\x01\"\xe7\x03\n\x0b\x45xecProfile\x12\x11\n\trun_count\x18\x01 \x01(\x03\x12\x18\n\x10\x61ll_start_micros\x18\x02 \x01(\x03\x12\x19\n\x11latest_end_micros\x18\x03 \x01(\x03\x12I\n\x11\x61\x63\x63\x65lerator_execs\x18\x04 \x03(\x0b\x32..tensorbored.ExecProfile.AcceleratorExecsEntry\x12\x39\n\tcpu_execs\x18\x05 \x03(\x0b\x32&.tensorbored.ExecProfile.CpuExecsEntry\x12-\n\x0cmemory_execs\x18\x07 \x03(\x0b\x32\x17.tensorbored.ExecMemory\x12\x32\n\x0b\x61llocations\x18\x0b \x03(\x0b\x32\x1d.tensorbored.AllocationRecord\x12\x0f\n\x07\x64\x65vices\x18\x06 \x03(\t\x1aN\n\x15\x41\x63\x63\x65leratorExecsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.tensorbored.ExecTime:\x02\x38\x01\x1a\x46\n\rCpuExecsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.tensorbored.ExecTime:\x02\x38\x01\"-\n\x08\x45xecTime\x12!\n\x05times\x18\x01 \x03(\x0b\x32\x12.tensorbored.Tuple\"\xa8\x03\n\nExecMemory\x12\x15\n\rmemory_micros\x18\x01 \x01(\x03\x12\x17\n\x0fhost_temp_bytes\x18\x02 \x01(\x03\x12\x1d\n\x15host_persistent_bytes\x18\x03 \x01(\x03\x12\x1e\n\x16\x61\x63\x63\x65lerator_temp_bytes\x18\x04 \x01(\x03\x12$\n\x1c\x61\x63\x63\x65lerator_persistent_bytes\x18\x05 \x01(\x03\x12\x17\n\x0frequested_bytes\x18\x06 \x01(\x03\x12\x12\n\npeak_bytes\x18\x07 \x01(\x03\x12\x16\n\x0eresidual_bytes\x18\x08 \x01(\x03\x12\x14\n\x0coutput_bytes\x18\t \x01(\x03\x12\x1e\n\x16\x61llocator_bytes_in_use\x18\n \x01(\x03\x12@\n\routput_memory\x18\x0b \x03(\x0b\x32).tensorbored.ExecMemory.OutputMemoryEntry\x1aH\n\x11OutputMemoryEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\"\n\x05value\x18\x02 \x01(\x0b\x32\x13.tensorbored.Memory:\x02\x38\x01\"\x1d\n\x05Tuple\x12\x14\n\x0cint64_values\x18\x01 \x03(\x03\"$\n\x06Memory\x12\r\n\x05\x62ytes\x18\x01 \x01(\x03\x12\x0b\n\x03ptr\x18\x02 \x01(\x04\x42RZPgithub.com/tensorflow/tensorflow/tensorflow/go/core/profiler/protos_all_go_protob\x06proto3')
+
+
+
+_CODEDEF = DESCRIPTOR.message_types_by_name['CodeDef']
+_CODEDEF_TRACE = _CODEDEF.nested_types_by_name['Trace']
+_OPLOGENTRY = DESCRIPTOR.message_types_by_name['OpLogEntry']
+_OPLOGPROTO = DESCRIPTOR.message_types_by_name['OpLogProto']
+_OPLOGPROTO_IDTOSTRINGENTRY = _OPLOGPROTO.nested_types_by_name['IdToStringEntry']
+_PROFILEPROTO = DESCRIPTOR.message_types_by_name['ProfileProto']
+_PROFILEPROTO_NODESENTRY = _PROFILEPROTO.nested_types_by_name['NodesEntry']
+_PROFILEPROTO_IDTOSTRINGENTRY = _PROFILEPROTO.nested_types_by_name['IdToStringEntry']
+_PROFILENODE = DESCRIPTOR.message_types_by_name['ProfileNode']
+_PROFILENODE_INPUTSENTRY = _PROFILENODE.nested_types_by_name['InputsEntry']
+_PROFILENODE_INPUTSHAPESENTRY = _PROFILENODE.nested_types_by_name['InputShapesEntry']
+_PROFILENODE_OUTPUTSENTRY = _PROFILENODE.nested_types_by_name['OutputsEntry']
+_PROFILENODE_OUTPUTSHAPESENTRY = _PROFILENODE.nested_types_by_name['OutputShapesEntry']
+_PROFILENODE_SRCOUTPUTINDEXENTRY = _PROFILENODE.nested_types_by_name['SrcOutputIndexEntry']
+_PROFILENODE_ATTRSENTRY = _PROFILENODE.nested_types_by_name['AttrsEntry']
+_PROFILENODE_EXECSENTRY = _PROFILENODE.nested_types_by_name['ExecsEntry']
+_EXECPROFILE = DESCRIPTOR.message_types_by_name['ExecProfile']
+_EXECPROFILE_ACCELERATOREXECSENTRY = _EXECPROFILE.nested_types_by_name['AcceleratorExecsEntry']
+_EXECPROFILE_CPUEXECSENTRY = _EXECPROFILE.nested_types_by_name['CpuExecsEntry']
+_EXECTIME = DESCRIPTOR.message_types_by_name['ExecTime']
+_EXECMEMORY = DESCRIPTOR.message_types_by_name['ExecMemory']
+_EXECMEMORY_OUTPUTMEMORYENTRY = _EXECMEMORY.nested_types_by_name['OutputMemoryEntry']
+_TUPLE = DESCRIPTOR.message_types_by_name['Tuple']
+_MEMORY = DESCRIPTOR.message_types_by_name['Memory']
+CodeDef = _reflection.GeneratedProtocolMessageType('CodeDef', (_message.Message,), {
+
+  'Trace' : _reflection.GeneratedProtocolMessageType('Trace', (_message.Message,), {
+    'DESCRIPTOR' : _CODEDEF_TRACE,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CodeDef.Trace)
+    })
+  ,
+  'DESCRIPTOR' : _CODEDEF,
+  '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.CodeDef)
+  })
+_sym_db.RegisterMessage(CodeDef)
+_sym_db.RegisterMessage(CodeDef.Trace)
+
+OpLogEntry = _reflection.GeneratedProtocolMessageType('OpLogEntry', (_message.Message,), {
+  'DESCRIPTOR' : _OPLOGENTRY,
+  '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.OpLogEntry)
+  })
+_sym_db.RegisterMessage(OpLogEntry)
+
+OpLogProto = _reflection.GeneratedProtocolMessageType('OpLogProto', (_message.Message,), {
+
+  'IdToStringEntry' : _reflection.GeneratedProtocolMessageType('IdToStringEntry', (_message.Message,), {
+    'DESCRIPTOR' : _OPLOGPROTO_IDTOSTRINGENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.OpLogProto.IdToStringEntry)
+    })
+  ,
+  'DESCRIPTOR' : _OPLOGPROTO,
+  '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.OpLogProto)
+  })
+_sym_db.RegisterMessage(OpLogProto)
+_sym_db.RegisterMessage(OpLogProto.IdToStringEntry)
+
+ProfileProto = _reflection.GeneratedProtocolMessageType('ProfileProto', (_message.Message,), {
+
+  'NodesEntry' : _reflection.GeneratedProtocolMessageType('NodesEntry', (_message.Message,), {
+    'DESCRIPTOR' : _PROFILEPROTO_NODESENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ProfileProto.NodesEntry)
+    })
+  ,
+
+  'IdToStringEntry' : _reflection.GeneratedProtocolMessageType('IdToStringEntry', (_message.Message,), {
+    'DESCRIPTOR' : _PROFILEPROTO_IDTOSTRINGENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ProfileProto.IdToStringEntry)
+    })
+  ,
+  'DESCRIPTOR' : _PROFILEPROTO,
+  '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ProfileProto)
+  })
+_sym_db.RegisterMessage(ProfileProto)
+_sym_db.RegisterMessage(ProfileProto.NodesEntry)
+_sym_db.RegisterMessage(ProfileProto.IdToStringEntry)
+
+ProfileNode = _reflection.GeneratedProtocolMessageType('ProfileNode', (_message.Message,), {
+
+  'InputsEntry' : _reflection.GeneratedProtocolMessageType('InputsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _PROFILENODE_INPUTSENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ProfileNode.InputsEntry)
+    })
+  ,
+
+  'InputShapesEntry' : _reflection.GeneratedProtocolMessageType('InputShapesEntry', (_message.Message,), {
+    'DESCRIPTOR' : _PROFILENODE_INPUTSHAPESENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ProfileNode.InputShapesEntry)
+    })
+  ,
+
+  'OutputsEntry' : _reflection.GeneratedProtocolMessageType('OutputsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _PROFILENODE_OUTPUTSENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ProfileNode.OutputsEntry)
+    })
+  ,
+
+  'OutputShapesEntry' : _reflection.GeneratedProtocolMessageType('OutputShapesEntry', (_message.Message,), {
+    'DESCRIPTOR' : _PROFILENODE_OUTPUTSHAPESENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ProfileNode.OutputShapesEntry)
+    })
+  ,
+
+  'SrcOutputIndexEntry' : _reflection.GeneratedProtocolMessageType('SrcOutputIndexEntry', (_message.Message,), {
+    'DESCRIPTOR' : _PROFILENODE_SRCOUTPUTINDEXENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ProfileNode.SrcOutputIndexEntry)
+    })
+  ,
+
+  'AttrsEntry' : _reflection.GeneratedProtocolMessageType('AttrsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _PROFILENODE_ATTRSENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ProfileNode.AttrsEntry)
+    })
+  ,
+
+  'ExecsEntry' : _reflection.GeneratedProtocolMessageType('ExecsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _PROFILENODE_EXECSENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ProfileNode.ExecsEntry)
+    })
+  ,
+  'DESCRIPTOR' : _PROFILENODE,
+  '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ProfileNode)
+  })
+_sym_db.RegisterMessage(ProfileNode)
+_sym_db.RegisterMessage(ProfileNode.InputsEntry)
+_sym_db.RegisterMessage(ProfileNode.InputShapesEntry)
+_sym_db.RegisterMessage(ProfileNode.OutputsEntry)
+_sym_db.RegisterMessage(ProfileNode.OutputShapesEntry)
+_sym_db.RegisterMessage(ProfileNode.SrcOutputIndexEntry)
+_sym_db.RegisterMessage(ProfileNode.AttrsEntry)
+_sym_db.RegisterMessage(ProfileNode.ExecsEntry)
+
+ExecProfile = _reflection.GeneratedProtocolMessageType('ExecProfile', (_message.Message,), {
+
+  'AcceleratorExecsEntry' : _reflection.GeneratedProtocolMessageType('AcceleratorExecsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _EXECPROFILE_ACCELERATOREXECSENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ExecProfile.AcceleratorExecsEntry)
+    })
+  ,
+
+  'CpuExecsEntry' : _reflection.GeneratedProtocolMessageType('CpuExecsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _EXECPROFILE_CPUEXECSENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ExecProfile.CpuExecsEntry)
+    })
+  ,
+  'DESCRIPTOR' : _EXECPROFILE,
+  '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ExecProfile)
+  })
+_sym_db.RegisterMessage(ExecProfile)
+_sym_db.RegisterMessage(ExecProfile.AcceleratorExecsEntry)
+_sym_db.RegisterMessage(ExecProfile.CpuExecsEntry)
+
+ExecTime = _reflection.GeneratedProtocolMessageType('ExecTime', (_message.Message,), {
+  'DESCRIPTOR' : _EXECTIME,
+  '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ExecTime)
+  })
+_sym_db.RegisterMessage(ExecTime)
+
+ExecMemory = _reflection.GeneratedProtocolMessageType('ExecMemory', (_message.Message,), {
+
+  'OutputMemoryEntry' : _reflection.GeneratedProtocolMessageType('OutputMemoryEntry', (_message.Message,), {
+    'DESCRIPTOR' : _EXECMEMORY_OUTPUTMEMORYENTRY,
+    '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ExecMemory.OutputMemoryEntry)
+    })
+  ,
+  'DESCRIPTOR' : _EXECMEMORY,
+  '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ExecMemory)
+  })
+_sym_db.RegisterMessage(ExecMemory)
+_sym_db.RegisterMessage(ExecMemory.OutputMemoryEntry)
+
+Tuple = _reflection.GeneratedProtocolMessageType('Tuple', (_message.Message,), {
+  'DESCRIPTOR' : _TUPLE,
+  '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.Tuple)
+  })
+_sym_db.RegisterMessage(Tuple)
+
+Memory = _reflection.GeneratedProtocolMessageType('Memory', (_message.Message,), {
+  'DESCRIPTOR' : _MEMORY,
+  '__module__' : 'tensorbored.compat.proto.tfprof_log_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.Memory)
+  })
+_sym_db.RegisterMessage(Memory)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'ZPgithub.com/tensorflow/tensorflow/tensorflow/go/core/profiler/protos_all_go_proto'
+  _OPLOGPROTO_IDTOSTRINGENTRY._options = None
+  _OPLOGPROTO_IDTOSTRINGENTRY._serialized_options = b'8\001'
+  _PROFILEPROTO_NODESENTRY._options = None
+  _PROFILEPROTO_NODESENTRY._serialized_options = b'8\001'
+  _PROFILEPROTO_IDTOSTRINGENTRY._options = None
+  _PROFILEPROTO_IDTOSTRINGENTRY._serialized_options = b'8\001'
+  _PROFILENODE_INPUTSENTRY._options = None
+  _PROFILENODE_INPUTSENTRY._serialized_options = b'8\001'
+  _PROFILENODE_INPUTSHAPESENTRY._options = None
+  _PROFILENODE_INPUTSHAPESENTRY._serialized_options = b'8\001'
+  _PROFILENODE_OUTPUTSENTRY._options = None
+  _PROFILENODE_OUTPUTSENTRY._serialized_options = b'8\001'
+  _PROFILENODE_OUTPUTSHAPESENTRY._options = None
+  _PROFILENODE_OUTPUTSHAPESENTRY._serialized_options = b'8\001'
+  _PROFILENODE_SRCOUTPUTINDEXENTRY._options = None
+  _PROFILENODE_SRCOUTPUTINDEXENTRY._serialized_options = b'8\001'
+  _PROFILENODE_ATTRSENTRY._options = None
+  _PROFILENODE_ATTRSENTRY._serialized_options = b'8\001'
+  _PROFILENODE_EXECSENTRY._options = None
+  _PROFILENODE_EXECSENTRY._serialized_options = b'8\001'
+  _EXECPROFILE_ACCELERATOREXECSENTRY._options = None
+  _EXECPROFILE_ACCELERATOREXECSENTRY._serialized_options = b'8\001'
+  _EXECPROFILE_CPUEXECSENTRY._options = None
+  _EXECPROFILE_CPUEXECSENTRY._serialized_options = b'8\001'
+  _EXECMEMORY_OUTPUTMEMORYENTRY._options = None
+  _EXECMEMORY_OUTPUTMEMORYENTRY._serialized_options = b'8\001'
+  _CODEDEF._serialized_start=145
+  _CODEDEF._serialized_end=344
+  _CODEDEF_TRACE._serialized_start=201
+  _CODEDEF_TRACE._serialized_end=344
+  _OPLOGENTRY._serialized_start=346
+  _OPLOGENTRY._serialized_end=446
+  _OPLOGPROTO._serialized_start=449
+  _OPLOGPROTO._serialized_end=621
+  _OPLOGPROTO_IDTOSTRINGENTRY._serialized_start=572
+  _OPLOGPROTO_IDTOSTRINGENTRY._serialized_end=621
+  _PROFILEPROTO._serialized_start=624
+  _PROFILEPROTO._serialized_end=946
+  _PROFILEPROTO_NODESENTRY._serialized_start=825
+  _PROFILEPROTO_NODESENTRY._serialized_end=895
+  _PROFILEPROTO_IDTOSTRINGENTRY._serialized_start=572
+  _PROFILEPROTO_IDTOSTRINGENTRY._serialized_end=621
+  _PROFILENODE._serialized_start=949
+  _PROFILENODE._serialized_end=1991
+  _PROFILENODE_INPUTSENTRY._serialized_start=1556
+  _PROFILENODE_INPUTSENTRY._serialized_end=1601
+  _PROFILENODE_INPUTSHAPESENTRY._serialized_start=1603
+  _PROFILENODE_INPUTSHAPESENTRY._serialized_end=1673
+  _PROFILENODE_OUTPUTSENTRY._serialized_start=1675
+  _PROFILENODE_OUTPUTSENTRY._serialized_end=1721
+  _PROFILENODE_OUTPUTSHAPESENTRY._serialized_start=1723
+  _PROFILENODE_OUTPUTSHAPESENTRY._serialized_end=1794
+  _PROFILENODE_SRCOUTPUTINDEXENTRY._serialized_start=1796
+  _PROFILENODE_SRCOUTPUTINDEXENTRY._serialized_end=1849
+  _PROFILENODE_ATTRSENTRY._serialized_start=1851
+  _PROFILENODE_ATTRSENTRY._serialized_end=1919
+  _PROFILENODE_EXECSENTRY._serialized_start=1921
+  _PROFILENODE_EXECSENTRY._serialized_end=1991
+  _EXECPROFILE._serialized_start=1994
+  _EXECPROFILE._serialized_end=2481
+  _EXECPROFILE_ACCELERATOREXECSENTRY._serialized_start=2331
+  _EXECPROFILE_ACCELERATOREXECSENTRY._serialized_end=2409
+  _EXECPROFILE_CPUEXECSENTRY._serialized_start=2411
+  _EXECPROFILE_CPUEXECSENTRY._serialized_end=2481
+  _EXECTIME._serialized_start=2483
+  _EXECTIME._serialized_end=2528
+  _EXECMEMORY._serialized_start=2531
+  _EXECMEMORY._serialized_end=2955
+  _EXECMEMORY_OUTPUTMEMORYENTRY._serialized_start=2883
+  _EXECMEMORY_OUTPUTMEMORYENTRY._serialized_end=2955
+  _TUPLE._serialized_start=2957
+  _TUPLE._serialized_end=2986
+  _MEMORY._serialized_start=2988
+  _MEMORY._serialized_end=3024
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/trackable_object_graph_pb2.py

@@ -0,0 +1,90 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/trackable_object_graph.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n5tensorbored/compat/proto/trackable_object_graph.proto\x12\x0btensorbored\x1a\x1egoogle/protobuf/wrappers.proto\"\xf8\x05\n\x14TrackableObjectGraph\x12@\n\x05nodes\x18\x01 \x03(\x0b\x32\x31.tensorbored.TrackableObjectGraph.TrackableObject\x1a\x9d\x05\n\x0fTrackableObject\x12S\n\x08\x63hildren\x18\x01 \x03(\x0b\x32\x41.tensorbored.TrackableObjectGraph.TrackableObject.ObjectReference\x12V\n\nattributes\x18\x02 \x03(\x0b\x32\x42.tensorbored.TrackableObjectGraph.TrackableObject.SerializedTensor\x12_\n\x0eslot_variables\x18\x03 \x03(\x0b\x32G.tensorbored.TrackableObjectGraph.TrackableObject.SlotVariableReference\x12\x36\n\x10registered_saver\x18\x04 \x01(\x0b\x32\x1c.tensorbored.RegisteredSaver\x12\x39\n\x15has_checkpoint_values\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.BoolValue\x1a\x36\n\x0fObjectReference\x12\x0f\n\x07node_id\x18\x01 \x01(\x05\x12\x12\n\nlocal_name\x18\x02 \x01(\t\x1a\x63\n\x10SerializedTensor\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\tfull_name\x18\x02 \x01(\t\x12\x16\n\x0e\x63heckpoint_key\x18\x03 \x01(\tJ\x04\x08\x04\x10\x05R\x10optional_restore\x1al\n\x15SlotVariableReference\x12!\n\x19original_variable_node_id\x18\x01 \x01(\x05\x12\x11\n\tslot_name\x18\x02 \x01(\t\x12\x1d\n\x15slot_variable_node_id\x18\x03 \x01(\x05\"4\n\x0fRegisteredSaver\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0bobject_name\x18\x02 \x01(\tBZZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_TRACKABLEOBJECTGRAPH = DESCRIPTOR.message_types_by_name['TrackableObjectGraph']
+_TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT = _TRACKABLEOBJECTGRAPH.nested_types_by_name['TrackableObject']
+_TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_OBJECTREFERENCE = _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT.nested_types_by_name['ObjectReference']
+_TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_SERIALIZEDTENSOR = _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT.nested_types_by_name['SerializedTensor']
+_TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_SLOTVARIABLEREFERENCE = _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT.nested_types_by_name['SlotVariableReference']
+_REGISTEREDSAVER = DESCRIPTOR.message_types_by_name['RegisteredSaver']
+TrackableObjectGraph = _reflection.GeneratedProtocolMessageType('TrackableObjectGraph', (_message.Message,), {
+
+  'TrackableObject' : _reflection.GeneratedProtocolMessageType('TrackableObject', (_message.Message,), {
+
+    'ObjectReference' : _reflection.GeneratedProtocolMessageType('ObjectReference', (_message.Message,), {
+      'DESCRIPTOR' : _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_OBJECTREFERENCE,
+      '__module__' : 'tensorbored.compat.proto.trackable_object_graph_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.TrackableObjectGraph.TrackableObject.ObjectReference)
+      })
+    ,
+
+    'SerializedTensor' : _reflection.GeneratedProtocolMessageType('SerializedTensor', (_message.Message,), {
+      'DESCRIPTOR' : _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_SERIALIZEDTENSOR,
+      '__module__' : 'tensorbored.compat.proto.trackable_object_graph_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.TrackableObjectGraph.TrackableObject.SerializedTensor)
+      })
+    ,
+
+    'SlotVariableReference' : _reflection.GeneratedProtocolMessageType('SlotVariableReference', (_message.Message,), {
+      'DESCRIPTOR' : _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_SLOTVARIABLEREFERENCE,
+      '__module__' : 'tensorbored.compat.proto.trackable_object_graph_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.TrackableObjectGraph.TrackableObject.SlotVariableReference)
+      })
+    ,
+    'DESCRIPTOR' : _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT,
+    '__module__' : 'tensorbored.compat.proto.trackable_object_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.TrackableObjectGraph.TrackableObject)
+    })
+  ,
+  'DESCRIPTOR' : _TRACKABLEOBJECTGRAPH,
+  '__module__' : 'tensorbored.compat.proto.trackable_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.TrackableObjectGraph)
+  })
+_sym_db.RegisterMessage(TrackableObjectGraph)
+_sym_db.RegisterMessage(TrackableObjectGraph.TrackableObject)
+_sym_db.RegisterMessage(TrackableObjectGraph.TrackableObject.ObjectReference)
+_sym_db.RegisterMessage(TrackableObjectGraph.TrackableObject.SerializedTensor)
+_sym_db.RegisterMessage(TrackableObjectGraph.TrackableObject.SlotVariableReference)
+
+RegisteredSaver = _reflection.GeneratedProtocolMessageType('RegisteredSaver', (_message.Message,), {
+  'DESCRIPTOR' : _REGISTEREDSAVER,
+  '__module__' : 'tensorbored.compat.proto.trackable_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.RegisteredSaver)
+  })
+_sym_db.RegisterMessage(RegisteredSaver)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\370\001\001'
+  _TRACKABLEOBJECTGRAPH._serialized_start=103
+  _TRACKABLEOBJECTGRAPH._serialized_end=863
+  _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT._serialized_start=194
+  _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT._serialized_end=863
+  _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_OBJECTREFERENCE._serialized_start=598
+  _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_OBJECTREFERENCE._serialized_end=652
+  _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_SERIALIZEDTENSOR._serialized_start=654
+  _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_SERIALIZEDTENSOR._serialized_end=753
+  _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_SLOTVARIABLEREFERENCE._serialized_start=755
+  _TRACKABLEOBJECTGRAPH_TRACKABLEOBJECT_SLOTVARIABLEREFERENCE._serialized_end=863
+  _REGISTEREDSAVER._serialized_start=865
+  _REGISTEREDSAVER._serialized_end=917
+# @@protoc_insertion_point(module_scope)