tensorbored 2.21.0rc1769983804__py3-none-any.whl

tensorbored/compat/proto/debug_pb2.py

@@ -0,0 +1,65 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/debug.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/debug.proto\x12\x0btensorbored\"\x8e\x01\n\x10\x44\x65\x62ugTensorWatch\x12\x11\n\tnode_name\x18\x01 \x01(\t\x12\x13\n\x0boutput_slot\x18\x02 \x01(\x05\x12\x11\n\tdebug_ops\x18\x03 \x03(\t\x12\x12\n\ndebug_urls\x18\x04 \x03(\t\x12+\n#tolerate_debug_op_creation_failures\x18\x05 \x01(\x08\"\x82\x01\n\x0c\x44\x65\x62ugOptions\x12>\n\x17\x64\x65\x62ug_tensor_watch_opts\x18\x04 \x03(\x0b\x32\x1d.tensorbored.DebugTensorWatch\x12\x13\n\x0bglobal_step\x18\n \x01(\x03\x12\x1d\n\x15reset_disk_byte_usage\x18\x0b \x01(\x08\"j\n\x12\x44\x65\x62uggedSourceFile\x12\x0c\n\x04host\x18\x01 \x01(\t\x12\x11\n\tfile_path\x18\x02 \x01(\t\x12\x15\n\rlast_modified\x18\x03 \x01(\x03\x12\r\n\x05\x62ytes\x18\x04 \x01(\x03\x12\r\n\x05lines\x18\x05 \x03(\t\"L\n\x13\x44\x65\x62uggedSourceFiles\x12\x35\n\x0csource_files\x18\x01 \x03(\x0b\x32\x1f.tensorbored.DebuggedSourceFileB\x83\x01\n\x18org.tensorflow.frameworkB\x0b\x44\x65\x62ugProtosP\x01ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_DEBUGTENSORWATCH = DESCRIPTOR.message_types_by_name['DebugTensorWatch']
+_DEBUGOPTIONS = DESCRIPTOR.message_types_by_name['DebugOptions']
+_DEBUGGEDSOURCEFILE = DESCRIPTOR.message_types_by_name['DebuggedSourceFile']
+_DEBUGGEDSOURCEFILES = DESCRIPTOR.message_types_by_name['DebuggedSourceFiles']
+DebugTensorWatch = _reflection.GeneratedProtocolMessageType('DebugTensorWatch', (_message.Message,), {
+  'DESCRIPTOR' : _DEBUGTENSORWATCH,
+  '__module__' : 'tensorbored.compat.proto.debug_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.DebugTensorWatch)
+  })
+_sym_db.RegisterMessage(DebugTensorWatch)
+
+DebugOptions = _reflection.GeneratedProtocolMessageType('DebugOptions', (_message.Message,), {
+  'DESCRIPTOR' : _DEBUGOPTIONS,
+  '__module__' : 'tensorbored.compat.proto.debug_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.DebugOptions)
+  })
+_sym_db.RegisterMessage(DebugOptions)
+
+DebuggedSourceFile = _reflection.GeneratedProtocolMessageType('DebuggedSourceFile', (_message.Message,), {
+  'DESCRIPTOR' : _DEBUGGEDSOURCEFILE,
+  '__module__' : 'tensorbored.compat.proto.debug_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.DebuggedSourceFile)
+  })
+_sym_db.RegisterMessage(DebuggedSourceFile)
+
+DebuggedSourceFiles = _reflection.GeneratedProtocolMessageType('DebuggedSourceFiles', (_message.Message,), {
+  'DESCRIPTOR' : _DEBUGGEDSOURCEFILES,
+  '__module__' : 'tensorbored.compat.proto.debug_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.DebuggedSourceFiles)
+  })
+_sym_db.RegisterMessage(DebuggedSourceFiles)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\013DebugProtosP\001ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\370\001\001'
+  _DEBUGTENSORWATCH._serialized_start=54
+  _DEBUGTENSORWATCH._serialized_end=196
+  _DEBUGOPTIONS._serialized_start=199
+  _DEBUGOPTIONS._serialized_end=329
+  _DEBUGGEDSOURCEFILE._serialized_start=331
+  _DEBUGGEDSOURCEFILE._serialized_end=437
+  _DEBUGGEDSOURCEFILES._serialized_start=439
+  _DEBUGGEDSOURCEFILES._serialized_end=515
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/event_pb2.py

@@ -0,0 +1,149 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/event.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+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 summary_pb2 as tensorbored_dot_compat_dot_proto_dot_summary__pb2
+try:
+  tensorbored_dot_compat_dot_proto_dot_histogram__pb2 = tensorbored_dot_compat_dot_proto_dot_summary__pb2.tensorbored_dot_compat_dot_proto_dot_histogram__pb2
+except AttributeError:
+  tensorbored_dot_compat_dot_proto_dot_histogram__pb2 = tensorbored_dot_compat_dot_proto_dot_summary__pb2.tensorbored.compat.proto.histogram_pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$tensorbored/compat/proto/event.proto\x12\x0btensorbored\x1a&tensorbored/compat/proto/summary.proto\"\xf9\x02\n\x05\x45vent\x12\x11\n\twall_time\x18\x01 \x01(\x01\x12\x0c\n\x04step\x18\x02 \x01(\x03\x12\x16\n\x0c\x66ile_version\x18\x03 \x01(\tH\x00\x12\x13\n\tgraph_def\x18\x04 \x01(\x0cH\x00\x12\'\n\x07summary\x18\x05 \x01(\x0b\x32\x14.tensorbored.SummaryH\x00\x12\x32\n\x0blog_message\x18\x06 \x01(\x0b\x32\x17.tensorbored.LogMessageB\x02\x18\x01H\x00\x12.\n\x0bsession_log\x18\x07 \x01(\x0b\x32\x17.tensorbored.SessionLogH\x00\x12=\n\x13tagged_run_metadata\x18\x08 \x01(\x0b\x32\x1e.tensorbored.TaggedRunMetadataH\x00\x12\x18\n\x0emeta_graph_def\x18\t \x01(\x0cH\x00\x12\x34\n\x0fsource_metadata\x18\n \x01(\x0b\x32\x1b.tensorbored.SourceMetadataB\x06\n\x04what\" \n\x0eSourceMetadata\x12\x0e\n\x06writer\x18\x01 \x01(\t\"\xa2\x01\n\nLogMessage\x12,\n\x05level\x18\x01 \x01(\x0e\x32\x1d.tensorbored.LogMessage.Level\x12\x0f\n\x07message\x18\x02 \x01(\t\"Q\n\x05Level\x12\x0b\n\x07UNKNOWN\x10\x00\x12\r\n\tDEBUGGING\x10\n\x12\x08\n\x04INFO\x10\x14\x12\x08\n\x04WARN\x10\x1e\x12\t\n\x05\x45RROR\x10(\x12\t\n\x05\x46\x41TAL\x10\x32\x1a\x02\x18\x01:\x02\x18\x01\"\xb7\x01\n\nSessionLog\x12\x35\n\x06status\x18\x01 \x01(\x0e\x32%.tensorbored.SessionLog.SessionStatus\x12\x17\n\x0f\x63heckpoint_path\x18\x02 \x01(\t\x12\x0b\n\x03msg\x18\x03 \x01(\t\"L\n\rSessionStatus\x12\x16\n\x12STATUS_UNSPECIFIED\x10\x00\x12\t\n\x05START\x10\x01\x12\x08\n\x04STOP\x10\x02\x12\x0e\n\nCHECKPOINT\x10\x03\"6\n\x11TaggedRunMetadata\x12\x0b\n\x03tag\x18\x01 \x01(\t\x12\x14\n\x0crun_metadata\x18\x02 \x01(\x0c\"$\n\x0eWatchdogConfig\x12\x12\n\ntimeout_ms\x18\x01 \x01(\x03\"&\n\x11RequestedExitCode\x12\x11\n\texit_code\x18\x01 \x01(\x05\"\xb9\x01\n\x16WorkerHeartbeatRequest\x12\x36\n\rshutdown_mode\x18\x01 \x01(\x0e\x32\x1f.tensorbored.WorkerShutdownMode\x12\x34\n\x0fwatchdog_config\x18\x02 \x01(\x0b\x32\x1b.tensorbored.WatchdogConfig\x12\x31\n\texit_code\x18\x03 \x01(\x0b\x32\x1e.tensorbored.RequestedExitCode\"\x85\x01\n\x17WorkerHeartbeatResponse\x12\x30\n\rhealth_status\x18\x01 \x01(\x0e\x32\x19.tensorbored.WorkerHealth\x12&\n\nworker_log\x18\x02 \x03(\x0b\x32\x12.tensorbored.Event\x12\x10\n\x08hostname\x18\x03 \x01(\t*[\n\x0cWorkerHealth\x12\x06\n\x02OK\x10\x00\x12\x1c\n\x18RECEIVED_SHUTDOWN_SIGNAL\x10\x01\x12\x12\n\x0eINTERNAL_ERROR\x10\x02\x12\x11\n\rSHUTTING_DOWN\x10\x03*k\n\x12WorkerShutdownMode\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x12\n\x0eNOT_CONFIGURED\x10\x01\x12\x18\n\x14WAIT_FOR_COORDINATOR\x10\x02\x12\x1a\n\x16SHUTDOWN_AFTER_TIMEOUT\x10\x03\x42p\n\x13org.tensorflow.utilB\x0b\x45ventProtosP\x01ZGgithub.com/tensorflow/tensorflow/tensorflow/go/core/util/event_go_proto\xf8\x01\x01\x62\x06proto3')
+
+_WORKERHEALTH = DESCRIPTOR.enum_types_by_name['WorkerHealth']
+WorkerHealth = enum_type_wrapper.EnumTypeWrapper(_WORKERHEALTH)
+_WORKERSHUTDOWNMODE = DESCRIPTOR.enum_types_by_name['WorkerShutdownMode']
+WorkerShutdownMode = enum_type_wrapper.EnumTypeWrapper(_WORKERSHUTDOWNMODE)
+OK = 0
+RECEIVED_SHUTDOWN_SIGNAL = 1
+INTERNAL_ERROR = 2
+SHUTTING_DOWN = 3
+DEFAULT = 0
+NOT_CONFIGURED = 1
+WAIT_FOR_COORDINATOR = 2
+SHUTDOWN_AFTER_TIMEOUT = 3
+
+
+_EVENT = DESCRIPTOR.message_types_by_name['Event']
+_SOURCEMETADATA = DESCRIPTOR.message_types_by_name['SourceMetadata']
+_LOGMESSAGE = DESCRIPTOR.message_types_by_name['LogMessage']
+_SESSIONLOG = DESCRIPTOR.message_types_by_name['SessionLog']
+_TAGGEDRUNMETADATA = DESCRIPTOR.message_types_by_name['TaggedRunMetadata']
+_WATCHDOGCONFIG = DESCRIPTOR.message_types_by_name['WatchdogConfig']
+_REQUESTEDEXITCODE = DESCRIPTOR.message_types_by_name['RequestedExitCode']
+_WORKERHEARTBEATREQUEST = DESCRIPTOR.message_types_by_name['WorkerHeartbeatRequest']
+_WORKERHEARTBEATRESPONSE = DESCRIPTOR.message_types_by_name['WorkerHeartbeatResponse']
+_LOGMESSAGE_LEVEL = _LOGMESSAGE.enum_types_by_name['Level']
+_SESSIONLOG_SESSIONSTATUS = _SESSIONLOG.enum_types_by_name['SessionStatus']
+Event = _reflection.GeneratedProtocolMessageType('Event', (_message.Message,), {
+  'DESCRIPTOR' : _EVENT,
+  '__module__' : 'tensorbored.compat.proto.event_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.Event)
+  })
+_sym_db.RegisterMessage(Event)
+
+SourceMetadata = _reflection.GeneratedProtocolMessageType('SourceMetadata', (_message.Message,), {
+  'DESCRIPTOR' : _SOURCEMETADATA,
+  '__module__' : 'tensorbored.compat.proto.event_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SourceMetadata)
+  })
+_sym_db.RegisterMessage(SourceMetadata)
+
+LogMessage = _reflection.GeneratedProtocolMessageType('LogMessage', (_message.Message,), {
+  'DESCRIPTOR' : _LOGMESSAGE,
+  '__module__' : 'tensorbored.compat.proto.event_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.LogMessage)
+  })
+_sym_db.RegisterMessage(LogMessage)
+
+SessionLog = _reflection.GeneratedProtocolMessageType('SessionLog', (_message.Message,), {
+  'DESCRIPTOR' : _SESSIONLOG,
+  '__module__' : 'tensorbored.compat.proto.event_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SessionLog)
+  })
+_sym_db.RegisterMessage(SessionLog)
+
+TaggedRunMetadata = _reflection.GeneratedProtocolMessageType('TaggedRunMetadata', (_message.Message,), {
+  'DESCRIPTOR' : _TAGGEDRUNMETADATA,
+  '__module__' : 'tensorbored.compat.proto.event_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.TaggedRunMetadata)
+  })
+_sym_db.RegisterMessage(TaggedRunMetadata)
+
+WatchdogConfig = _reflection.GeneratedProtocolMessageType('WatchdogConfig', (_message.Message,), {
+  'DESCRIPTOR' : _WATCHDOGCONFIG,
+  '__module__' : 'tensorbored.compat.proto.event_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.WatchdogConfig)
+  })
+_sym_db.RegisterMessage(WatchdogConfig)
+
+RequestedExitCode = _reflection.GeneratedProtocolMessageType('RequestedExitCode', (_message.Message,), {
+  'DESCRIPTOR' : _REQUESTEDEXITCODE,
+  '__module__' : 'tensorbored.compat.proto.event_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.RequestedExitCode)
+  })
+_sym_db.RegisterMessage(RequestedExitCode)
+
+WorkerHeartbeatRequest = _reflection.GeneratedProtocolMessageType('WorkerHeartbeatRequest', (_message.Message,), {
+  'DESCRIPTOR' : _WORKERHEARTBEATREQUEST,
+  '__module__' : 'tensorbored.compat.proto.event_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.WorkerHeartbeatRequest)
+  })
+_sym_db.RegisterMessage(WorkerHeartbeatRequest)
+
+WorkerHeartbeatResponse = _reflection.GeneratedProtocolMessageType('WorkerHeartbeatResponse', (_message.Message,), {
+  'DESCRIPTOR' : _WORKERHEARTBEATRESPONSE,
+  '__module__' : 'tensorbored.compat.proto.event_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.WorkerHeartbeatResponse)
+  })
+_sym_db.RegisterMessage(WorkerHeartbeatResponse)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\023org.tensorflow.utilB\013EventProtosP\001ZGgithub.com/tensorflow/tensorflow/tensorflow/go/core/util/event_go_proto\370\001\001'
+  _EVENT.fields_by_name['log_message']._options = None
+  _EVENT.fields_by_name['log_message']._serialized_options = b'\030\001'
+  _LOGMESSAGE_LEVEL._options = None
+  _LOGMESSAGE_LEVEL._serialized_options = b'\030\001'
+  _LOGMESSAGE._options = None
+  _LOGMESSAGE._serialized_options = b'\030\001'
+  _WORKERHEALTH._serialized_start=1316
+  _WORKERHEALTH._serialized_end=1407
+  _WORKERSHUTDOWNMODE._serialized_start=1409
+  _WORKERSHUTDOWNMODE._serialized_end=1516
+  _EVENT._serialized_start=94
+  _EVENT._serialized_end=471
+  _SOURCEMETADATA._serialized_start=473
+  _SOURCEMETADATA._serialized_end=505
+  _LOGMESSAGE._serialized_start=508
+  _LOGMESSAGE._serialized_end=670
+  _LOGMESSAGE_LEVEL._serialized_start=585
+  _LOGMESSAGE_LEVEL._serialized_end=666
+  _SESSIONLOG._serialized_start=673
+  _SESSIONLOG._serialized_end=856
+  _SESSIONLOG_SESSIONSTATUS._serialized_start=780
+  _SESSIONLOG_SESSIONSTATUS._serialized_end=856
+  _TAGGEDRUNMETADATA._serialized_start=858
+  _TAGGEDRUNMETADATA._serialized_end=912
+  _WATCHDOGCONFIG._serialized_start=914
+  _WATCHDOGCONFIG._serialized_end=950
+  _REQUESTEDEXITCODE._serialized_start=952
+  _REQUESTEDEXITCODE._serialized_end=990
+  _WORKERHEARTBEATREQUEST._serialized_start=993
+  _WORKERHEARTBEATREQUEST._serialized_end=1178
+  _WORKERHEARTBEATRESPONSE._serialized_start=1181
+  _WORKERHEARTBEATRESPONSE._serialized_end=1314
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/full_type_pb2.py

@@ -0,0 +1,74 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/full_type.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+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/full_type.proto\x12\x0btensorbored\"\x81\x01\n\x0b\x46ullTypeDef\x12(\n\x07type_id\x18\x01 \x01(\x0e\x32\x17.tensorbored.FullTypeId\x12&\n\x04\x61rgs\x18\x02 \x03(\x0b\x32\x18.tensorbored.FullTypeDef\x12\x0b\n\x01s\x18\x03 \x01(\tH\x00\x12\x0b\n\x01i\x18\x04 \x01(\x03H\x00\x42\x06\n\x04\x61ttr*\xda\x04\n\nFullTypeId\x12\r\n\tTFT_UNSET\x10\x00\x12\x0b\n\x07TFT_VAR\x10\x01\x12\x0b\n\x07TFT_ANY\x10\x02\x12\x0f\n\x0bTFT_PRODUCT\x10\x03\x12\r\n\tTFT_NAMED\x10\x04\x12\x10\n\x0cTFT_FOR_EACH\x10\x14\x12\x10\n\x0cTFT_CALLABLE\x10\x64\x12\x0f\n\nTFT_TENSOR\x10\xe8\x07\x12\x0e\n\tTFT_ARRAY\x10\xe9\x07\x12\x11\n\x0cTFT_OPTIONAL\x10\xea\x07\x12\x10\n\x0bTFT_LITERAL\x10\xeb\x07\x12\x10\n\x0bTFT_ENCODED\x10\xec\x07\x12\x15\n\x10TFT_SHAPE_TENSOR\x10\xed\x07\x12\r\n\x08TFT_BOOL\x10\xc8\x01\x12\x0e\n\tTFT_UINT8\x10\xc9\x01\x12\x0f\n\nTFT_UINT16\x10\xca\x01\x12\x0f\n\nTFT_UINT32\x10\xcb\x01\x12\x0f\n\nTFT_UINT64\x10\xcc\x01\x12\r\n\x08TFT_INT8\x10\xcd\x01\x12\x0e\n\tTFT_INT16\x10\xce\x01\x12\x0e\n\tTFT_INT32\x10\xcf\x01\x12\x0e\n\tTFT_INT64\x10\xd0\x01\x12\r\n\x08TFT_HALF\x10\xd1\x01\x12\x0e\n\tTFT_FLOAT\x10\xd2\x01\x12\x0f\n\nTFT_DOUBLE\x10\xd3\x01\x12\x11\n\x0cTFT_BFLOAT16\x10\xd7\x01\x12\x12\n\rTFT_COMPLEX64\x10\xd4\x01\x12\x13\n\x0eTFT_COMPLEX128\x10\xd5\x01\x12\x0f\n\nTFT_STRING\x10\xd6\x01\x12\x10\n\x0bTFT_DATASET\x10\xf6N\x12\x0f\n\nTFT_RAGGED\x10\xf7N\x12\x11\n\x0cTFT_ITERATOR\x10\xf8N\x12\x13\n\x0eTFT_MUTEX_LOCK\x10\xdaO\x12\x17\n\x12TFT_LEGACY_VARIANT\x10\xdbOB\x81\x01\n\x18org.tensorflow.frameworkB\x0e\x46ullTypeProtosP\x01ZPgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/full_type_go_proto\xf8\x01\x01\x62\x06proto3')
+
+_FULLTYPEID = DESCRIPTOR.enum_types_by_name['FullTypeId']
+FullTypeId = enum_type_wrapper.EnumTypeWrapper(_FULLTYPEID)
+TFT_UNSET = 0
+TFT_VAR = 1
+TFT_ANY = 2
+TFT_PRODUCT = 3
+TFT_NAMED = 4
+TFT_FOR_EACH = 20
+TFT_CALLABLE = 100
+TFT_TENSOR = 1000
+TFT_ARRAY = 1001
+TFT_OPTIONAL = 1002
+TFT_LITERAL = 1003
+TFT_ENCODED = 1004
+TFT_SHAPE_TENSOR = 1005
+TFT_BOOL = 200
+TFT_UINT8 = 201
+TFT_UINT16 = 202
+TFT_UINT32 = 203
+TFT_UINT64 = 204
+TFT_INT8 = 205
+TFT_INT16 = 206
+TFT_INT32 = 207
+TFT_INT64 = 208
+TFT_HALF = 209
+TFT_FLOAT = 210
+TFT_DOUBLE = 211
+TFT_BFLOAT16 = 215
+TFT_COMPLEX64 = 212
+TFT_COMPLEX128 = 213
+TFT_STRING = 214
+TFT_DATASET = 10102
+TFT_RAGGED = 10103
+TFT_ITERATOR = 10104
+TFT_MUTEX_LOCK = 10202
+TFT_LEGACY_VARIANT = 10203
+
+
+_FULLTYPEDEF = DESCRIPTOR.message_types_by_name['FullTypeDef']
+FullTypeDef = _reflection.GeneratedProtocolMessageType('FullTypeDef', (_message.Message,), {
+  'DESCRIPTOR' : _FULLTYPEDEF,
+  '__module__' : 'tensorbored.compat.proto.full_type_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.FullTypeDef)
+  })
+_sym_db.RegisterMessage(FullTypeDef)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\016FullTypeProtosP\001ZPgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/full_type_go_proto\370\001\001'
+  _FULLTYPEID._serialized_start=190
+  _FULLTYPEID._serialized_end=792
+  _FULLTYPEDEF._serialized_start=58
+  _FULLTYPEDEF._serialized_end=187
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/function_pb2.py

@@ -0,0 +1,157 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/function.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 node_def_pb2 as tensorbored_dot_compat_dot_proto_dot_node__def__pb2
+from tensorbored.compat.proto import op_def_pb2 as tensorbored_dot_compat_dot_proto_dot_op__def__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'tensorbored/compat/proto/function.proto\x12\x0btensorbored\x1a)tensorbored/compat/proto/attr_value.proto\x1a\'tensorbored/compat/proto/node_def.proto\x1a%tensorbored/compat/proto/op_def.proto\"\xab\x01\n\x12\x46unctionDefLibrary\x12*\n\x08\x66unction\x18\x01 \x03(\x0b\x32\x18.tensorbored.FunctionDef\x12*\n\x08gradient\x18\x02 \x03(\x0b\x32\x18.tensorbored.GradientDef\x12=\n\x14registered_gradients\x18\x03 \x03(\x0b\x32\x1f.tensorbored.RegisteredGradient\"\xcf\x06\n\x0b\x46unctionDef\x12%\n\tsignature\x18\x01 \x01(\x0b\x32\x12.tensorbored.OpDef\x12\x30\n\x04\x61ttr\x18\x05 \x03(\x0b\x32\".tensorbored.FunctionDef.AttrEntry\x12\x37\n\x08\x61rg_attr\x18\x07 \x03(\x0b\x32%.tensorbored.FunctionDef.ArgAttrEntry\x12Q\n\x16resource_arg_unique_id\x18\x08 \x03(\x0b\x32\x31.tensorbored.FunctionDef.ResourceArgUniqueIdEntry\x12&\n\x08node_def\x18\x03 \x03(\x0b\x32\x14.tensorbored.NodeDef\x12.\n\x03ret\x18\x04 \x03(\x0b\x32!.tensorbored.FunctionDef.RetEntry\x12=\n\x0b\x63ontrol_ret\x18\x06 \x03(\x0b\x32(.tensorbored.FunctionDef.ControlRetEntry\x1a\x43\n\tAttrEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.tensorbored.AttrValue:\x02\x38\x01\x1a\x8a\x01\n\x08\x41rgAttrs\x12\x39\n\x04\x61ttr\x18\x01 \x03(\x0b\x32+.tensorbored.FunctionDef.ArgAttrs.AttrEntry\x1a\x43\n\tAttrEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.tensorbored.AttrValue:\x02\x38\x01\x1aQ\n\x0c\x41rgAttrEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32!.tensorbored.FunctionDef.ArgAttrs:\x02\x38\x01\x1a:\n\x18ResourceArgUniqueIdEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12\r\n\x05value\x18\x02 \x01(\r:\x02\x38\x01\x1a*\n\x08RetEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\x31\n\x0f\x43ontrolRetEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01J\x04\x08\x02\x10\x03\";\n\x0bGradientDef\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x15\n\rgradient_func\x18\x02 \x01(\t\"G\n\x12RegisteredGradient\x12\x15\n\rgradient_func\x18\x01 \x01(\t\x12\x1a\n\x12registered_op_type\x18\x02 \x01(\tB\x80\x01\n\x18org.tensorflow.frameworkB\x0e\x46unctionProtosP\x01ZOgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/function_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_FUNCTIONDEFLIBRARY = DESCRIPTOR.message_types_by_name['FunctionDefLibrary']
+_FUNCTIONDEF = DESCRIPTOR.message_types_by_name['FunctionDef']
+_FUNCTIONDEF_ATTRENTRY = _FUNCTIONDEF.nested_types_by_name['AttrEntry']
+_FUNCTIONDEF_ARGATTRS = _FUNCTIONDEF.nested_types_by_name['ArgAttrs']
+_FUNCTIONDEF_ARGATTRS_ATTRENTRY = _FUNCTIONDEF_ARGATTRS.nested_types_by_name['AttrEntry']
+_FUNCTIONDEF_ARGATTRENTRY = _FUNCTIONDEF.nested_types_by_name['ArgAttrEntry']
+_FUNCTIONDEF_RESOURCEARGUNIQUEIDENTRY = _FUNCTIONDEF.nested_types_by_name['ResourceArgUniqueIdEntry']
+_FUNCTIONDEF_RETENTRY = _FUNCTIONDEF.nested_types_by_name['RetEntry']
+_FUNCTIONDEF_CONTROLRETENTRY = _FUNCTIONDEF.nested_types_by_name['ControlRetEntry']
+_GRADIENTDEF = DESCRIPTOR.message_types_by_name['GradientDef']
+_REGISTEREDGRADIENT = DESCRIPTOR.message_types_by_name['RegisteredGradient']
+FunctionDefLibrary = _reflection.GeneratedProtocolMessageType('FunctionDefLibrary', (_message.Message,), {
+  'DESCRIPTOR' : _FUNCTIONDEFLIBRARY,
+  '__module__' : 'tensorbored.compat.proto.function_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.FunctionDefLibrary)
+  })
+_sym_db.RegisterMessage(FunctionDefLibrary)
+
+FunctionDef = _reflection.GeneratedProtocolMessageType('FunctionDef', (_message.Message,), {
+
+  'AttrEntry' : _reflection.GeneratedProtocolMessageType('AttrEntry', (_message.Message,), {
+    'DESCRIPTOR' : _FUNCTIONDEF_ATTRENTRY,
+    '__module__' : 'tensorbored.compat.proto.function_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.FunctionDef.AttrEntry)
+    })
+  ,
+
+  'ArgAttrs' : _reflection.GeneratedProtocolMessageType('ArgAttrs', (_message.Message,), {
+
+    'AttrEntry' : _reflection.GeneratedProtocolMessageType('AttrEntry', (_message.Message,), {
+      'DESCRIPTOR' : _FUNCTIONDEF_ARGATTRS_ATTRENTRY,
+      '__module__' : 'tensorbored.compat.proto.function_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.FunctionDef.ArgAttrs.AttrEntry)
+      })
+    ,
+    'DESCRIPTOR' : _FUNCTIONDEF_ARGATTRS,
+    '__module__' : 'tensorbored.compat.proto.function_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.FunctionDef.ArgAttrs)
+    })
+  ,
+
+  'ArgAttrEntry' : _reflection.GeneratedProtocolMessageType('ArgAttrEntry', (_message.Message,), {
+    'DESCRIPTOR' : _FUNCTIONDEF_ARGATTRENTRY,
+    '__module__' : 'tensorbored.compat.proto.function_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.FunctionDef.ArgAttrEntry)
+    })
+  ,
+
+  'ResourceArgUniqueIdEntry' : _reflection.GeneratedProtocolMessageType('ResourceArgUniqueIdEntry', (_message.Message,), {
+    'DESCRIPTOR' : _FUNCTIONDEF_RESOURCEARGUNIQUEIDENTRY,
+    '__module__' : 'tensorbored.compat.proto.function_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.FunctionDef.ResourceArgUniqueIdEntry)
+    })
+  ,
+
+  'RetEntry' : _reflection.GeneratedProtocolMessageType('RetEntry', (_message.Message,), {
+    'DESCRIPTOR' : _FUNCTIONDEF_RETENTRY,
+    '__module__' : 'tensorbored.compat.proto.function_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.FunctionDef.RetEntry)
+    })
+  ,
+
+  'ControlRetEntry' : _reflection.GeneratedProtocolMessageType('ControlRetEntry', (_message.Message,), {
+    'DESCRIPTOR' : _FUNCTIONDEF_CONTROLRETENTRY,
+    '__module__' : 'tensorbored.compat.proto.function_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.FunctionDef.ControlRetEntry)
+    })
+  ,
+  'DESCRIPTOR' : _FUNCTIONDEF,
+  '__module__' : 'tensorbored.compat.proto.function_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.FunctionDef)
+  })
+_sym_db.RegisterMessage(FunctionDef)
+_sym_db.RegisterMessage(FunctionDef.AttrEntry)
+_sym_db.RegisterMessage(FunctionDef.ArgAttrs)
+_sym_db.RegisterMessage(FunctionDef.ArgAttrs.AttrEntry)
+_sym_db.RegisterMessage(FunctionDef.ArgAttrEntry)
+_sym_db.RegisterMessage(FunctionDef.ResourceArgUniqueIdEntry)
+_sym_db.RegisterMessage(FunctionDef.RetEntry)
+_sym_db.RegisterMessage(FunctionDef.ControlRetEntry)
+
+GradientDef = _reflection.GeneratedProtocolMessageType('GradientDef', (_message.Message,), {
+  'DESCRIPTOR' : _GRADIENTDEF,
+  '__module__' : 'tensorbored.compat.proto.function_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.GradientDef)
+  })
+_sym_db.RegisterMessage(GradientDef)
+
+RegisteredGradient = _reflection.GeneratedProtocolMessageType('RegisteredGradient', (_message.Message,), {
+  'DESCRIPTOR' : _REGISTEREDGRADIENT,
+  '__module__' : 'tensorbored.compat.proto.function_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.RegisteredGradient)
+  })
+_sym_db.RegisterMessage(RegisteredGradient)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\016FunctionProtosP\001ZOgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/function_go_proto\370\001\001'
+  _FUNCTIONDEF_ATTRENTRY._options = None
+  _FUNCTIONDEF_ATTRENTRY._serialized_options = b'8\001'
+  _FUNCTIONDEF_ARGATTRS_ATTRENTRY._options = None
+  _FUNCTIONDEF_ARGATTRS_ATTRENTRY._serialized_options = b'8\001'
+  _FUNCTIONDEF_ARGATTRENTRY._options = None
+  _FUNCTIONDEF_ARGATTRENTRY._serialized_options = b'8\001'
+  _FUNCTIONDEF_RESOURCEARGUNIQUEIDENTRY._options = None
+  _FUNCTIONDEF_RESOURCEARGUNIQUEIDENTRY._serialized_options = b'8\001'
+  _FUNCTIONDEF_RETENTRY._options = None
+  _FUNCTIONDEF_RETENTRY._serialized_options = b'8\001'
+  _FUNCTIONDEF_CONTROLRETENTRY._options = None
+  _FUNCTIONDEF_CONTROLRETENTRY._serialized_options = b'8\001'
+  _FUNCTIONDEFLIBRARY._serialized_start=180
+  _FUNCTIONDEFLIBRARY._serialized_end=351
+  _FUNCTIONDEF._serialized_start=354
+  _FUNCTIONDEF._serialized_end=1201
+  _FUNCTIONDEF_ATTRENTRY._serialized_start=749
+  _FUNCTIONDEF_ATTRENTRY._serialized_end=816
+  _FUNCTIONDEF_ARGATTRS._serialized_start=819
+  _FUNCTIONDEF_ARGATTRS._serialized_end=957
+  _FUNCTIONDEF_ARGATTRS_ATTRENTRY._serialized_start=749
+  _FUNCTIONDEF_ARGATTRS_ATTRENTRY._serialized_end=816
+  _FUNCTIONDEF_ARGATTRENTRY._serialized_start=959
+  _FUNCTIONDEF_ARGATTRENTRY._serialized_end=1040
+  _FUNCTIONDEF_RESOURCEARGUNIQUEIDENTRY._serialized_start=1042
+  _FUNCTIONDEF_RESOURCEARGUNIQUEIDENTRY._serialized_end=1100
+  _FUNCTIONDEF_RETENTRY._serialized_start=1102
+  _FUNCTIONDEF_RETENTRY._serialized_end=1144
+  _FUNCTIONDEF_CONTROLRETENTRY._serialized_start=1146
+  _FUNCTIONDEF_CONTROLRETENTRY._serialized_end=1195
+  _GRADIENTDEF._serialized_start=1203
+  _GRADIENTDEF._serialized_end=1262
+  _REGISTEREDGRADIENT._serialized_start=1264
+  _REGISTEREDGRADIENT._serialized_end=1335
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/graph_debug_info_pb2.py

@@ -0,0 +1,111 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/graph_debug_info.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/graph_debug_info.proto\x12\x0btensorbored\"\xab\x06\n\x0eGraphDebugInfo\x12\r\n\x05\x66iles\x18\x01 \x03(\t\x12\x41\n\x0c\x66rames_by_id\x18\x04 \x03(\x0b\x32+.tensorbored.GraphDebugInfo.FramesByIdEntry\x12\x41\n\x0ctraces_by_id\x18\x06 \x03(\x0b\x32+.tensorbored.GraphDebugInfo.TracesByIdEntry\x12\x37\n\x06traces\x18\x02 \x03(\x0b\x32\'.tensorbored.GraphDebugInfo.TracesEntry\x12H\n\x10name_to_trace_id\x18\x05 \x03(\x0b\x32..tensorbored.GraphDebugInfo.NameToTraceIdEntry\x1aX\n\x0b\x46ileLineCol\x12\x12\n\nfile_index\x18\x01 \x01(\x05\x12\x0c\n\x04line\x18\x02 \x01(\x05\x12\x0b\n\x03\x63ol\x18\x03 \x01(\x05\x12\x0c\n\x04\x66unc\x18\x04 \x01(\t\x12\x0c\n\x04\x63ode\x18\x05 \x01(\t\x1a\x63\n\nStackTrace\x12?\n\x0e\x66ile_line_cols\x18\x01 \x03(\x0b\x32\'.tensorbored.GraphDebugInfo.FileLineCol\x12\x14\n\x08\x66rame_id\x18\x02 \x03(\x06\x42\x02\x10\x01\x1aZ\n\x0f\x46ramesByIdEntry\x12\x0b\n\x03key\x18\x01 \x01(\x06\x12\x36\n\x05value\x18\x02 \x01(\x0b\x32\'.tensorbored.GraphDebugInfo.FileLineCol:\x02\x38\x01\x1aY\n\x0fTracesByIdEntry\x12\x0b\n\x03key\x18\x01 \x01(\x06\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32&.tensorbored.GraphDebugInfo.StackTrace:\x02\x38\x01\x1aU\n\x0bTracesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32&.tensorbored.GraphDebugInfo.StackTrace:\x02\x38\x01\x1a\x34\n\x12NameToTraceIdEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x06:\x02\x38\x01\x42\x8c\x01\n\x18org.tensorflow.frameworkB\x14GraphDebugInfoProtosP\x01ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\xf8\x01\x01')
+
+
+
+_GRAPHDEBUGINFO = DESCRIPTOR.message_types_by_name['GraphDebugInfo']
+_GRAPHDEBUGINFO_FILELINECOL = _GRAPHDEBUGINFO.nested_types_by_name['FileLineCol']
+_GRAPHDEBUGINFO_STACKTRACE = _GRAPHDEBUGINFO.nested_types_by_name['StackTrace']
+_GRAPHDEBUGINFO_FRAMESBYIDENTRY = _GRAPHDEBUGINFO.nested_types_by_name['FramesByIdEntry']
+_GRAPHDEBUGINFO_TRACESBYIDENTRY = _GRAPHDEBUGINFO.nested_types_by_name['TracesByIdEntry']
+_GRAPHDEBUGINFO_TRACESENTRY = _GRAPHDEBUGINFO.nested_types_by_name['TracesEntry']
+_GRAPHDEBUGINFO_NAMETOTRACEIDENTRY = _GRAPHDEBUGINFO.nested_types_by_name['NameToTraceIdEntry']
+GraphDebugInfo = _reflection.GeneratedProtocolMessageType('GraphDebugInfo', (_message.Message,), {
+
+  'FileLineCol' : _reflection.GeneratedProtocolMessageType('FileLineCol', (_message.Message,), {
+    'DESCRIPTOR' : _GRAPHDEBUGINFO_FILELINECOL,
+    '__module__' : 'tensorbored.compat.proto.graph_debug_info_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.GraphDebugInfo.FileLineCol)
+    })
+  ,
+
+  'StackTrace' : _reflection.GeneratedProtocolMessageType('StackTrace', (_message.Message,), {
+    'DESCRIPTOR' : _GRAPHDEBUGINFO_STACKTRACE,
+    '__module__' : 'tensorbored.compat.proto.graph_debug_info_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.GraphDebugInfo.StackTrace)
+    })
+  ,
+
+  'FramesByIdEntry' : _reflection.GeneratedProtocolMessageType('FramesByIdEntry', (_message.Message,), {
+    'DESCRIPTOR' : _GRAPHDEBUGINFO_FRAMESBYIDENTRY,
+    '__module__' : 'tensorbored.compat.proto.graph_debug_info_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.GraphDebugInfo.FramesByIdEntry)
+    })
+  ,
+
+  'TracesByIdEntry' : _reflection.GeneratedProtocolMessageType('TracesByIdEntry', (_message.Message,), {
+    'DESCRIPTOR' : _GRAPHDEBUGINFO_TRACESBYIDENTRY,
+    '__module__' : 'tensorbored.compat.proto.graph_debug_info_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.GraphDebugInfo.TracesByIdEntry)
+    })
+  ,
+
+  'TracesEntry' : _reflection.GeneratedProtocolMessageType('TracesEntry', (_message.Message,), {
+    'DESCRIPTOR' : _GRAPHDEBUGINFO_TRACESENTRY,
+    '__module__' : 'tensorbored.compat.proto.graph_debug_info_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.GraphDebugInfo.TracesEntry)
+    })
+  ,
+
+  'NameToTraceIdEntry' : _reflection.GeneratedProtocolMessageType('NameToTraceIdEntry', (_message.Message,), {
+    'DESCRIPTOR' : _GRAPHDEBUGINFO_NAMETOTRACEIDENTRY,
+    '__module__' : 'tensorbored.compat.proto.graph_debug_info_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.GraphDebugInfo.NameToTraceIdEntry)
+    })
+  ,
+  'DESCRIPTOR' : _GRAPHDEBUGINFO,
+  '__module__' : 'tensorbored.compat.proto.graph_debug_info_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.GraphDebugInfo)
+  })
+_sym_db.RegisterMessage(GraphDebugInfo)
+_sym_db.RegisterMessage(GraphDebugInfo.FileLineCol)
+_sym_db.RegisterMessage(GraphDebugInfo.StackTrace)
+_sym_db.RegisterMessage(GraphDebugInfo.FramesByIdEntry)
+_sym_db.RegisterMessage(GraphDebugInfo.TracesByIdEntry)
+_sym_db.RegisterMessage(GraphDebugInfo.TracesEntry)
+_sym_db.RegisterMessage(GraphDebugInfo.NameToTraceIdEntry)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\024GraphDebugInfoProtosP\001ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\370\001\001'
+  _GRAPHDEBUGINFO_STACKTRACE.fields_by_name['frame_id']._options = None
+  _GRAPHDEBUGINFO_STACKTRACE.fields_by_name['frame_id']._serialized_options = b'\020\001'
+  _GRAPHDEBUGINFO_FRAMESBYIDENTRY._options = None
+  _GRAPHDEBUGINFO_FRAMESBYIDENTRY._serialized_options = b'8\001'
+  _GRAPHDEBUGINFO_TRACESBYIDENTRY._options = None
+  _GRAPHDEBUGINFO_TRACESBYIDENTRY._serialized_options = b'8\001'
+  _GRAPHDEBUGINFO_TRACESENTRY._options = None
+  _GRAPHDEBUGINFO_TRACESENTRY._serialized_options = b'8\001'
+  _GRAPHDEBUGINFO_NAMETOTRACEIDENTRY._options = None
+  _GRAPHDEBUGINFO_NAMETOTRACEIDENTRY._serialized_options = b'8\001'
+  _GRAPHDEBUGINFO._serialized_start=65
+  _GRAPHDEBUGINFO._serialized_end=876
+  _GRAPHDEBUGINFO_FILELINECOL._serialized_start=363
+  _GRAPHDEBUGINFO_FILELINECOL._serialized_end=451
+  _GRAPHDEBUGINFO_STACKTRACE._serialized_start=453
+  _GRAPHDEBUGINFO_STACKTRACE._serialized_end=552
+  _GRAPHDEBUGINFO_FRAMESBYIDENTRY._serialized_start=554
+  _GRAPHDEBUGINFO_FRAMESBYIDENTRY._serialized_end=644
+  _GRAPHDEBUGINFO_TRACESBYIDENTRY._serialized_start=646
+  _GRAPHDEBUGINFO_TRACESBYIDENTRY._serialized_end=735
+  _GRAPHDEBUGINFO_TRACESENTRY._serialized_start=737
+  _GRAPHDEBUGINFO_TRACESENTRY._serialized_end=822
+  _GRAPHDEBUGINFO_NAMETOTRACEIDENTRY._serialized_start=824
+  _GRAPHDEBUGINFO_NAMETOTRACEIDENTRY._serialized_end=876
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/graph_pb2.py

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/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 tensorbored.compat.proto import function_pb2 as tensorbored_dot_compat_dot_proto_dot_function__pb2
+from tensorbored.compat.proto import graph_debug_info_pb2 as tensorbored_dot_compat_dot_proto_dot_graph__debug__info__pb2
+from tensorbored.compat.proto import node_def_pb2 as tensorbored_dot_compat_dot_proto_dot_node__def__pb2
+from tensorbored.compat.proto import versions_pb2 as tensorbored_dot_compat_dot_proto_dot_versions__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$tensorbored/compat/proto/graph.proto\x12\x0btensorbored\x1a\'tensorbored/compat/proto/function.proto\x1a/tensorbored/compat/proto/graph_debug_info.proto\x1a\'tensorbored/compat/proto/node_def.proto\x1a\'tensorbored/compat/proto/versions.proto\"\xd1\x01\n\x08GraphDef\x12\"\n\x04node\x18\x01 \x03(\x0b\x32\x14.tensorbored.NodeDef\x12)\n\x08versions\x18\x04 \x01(\x0b\x32\x17.tensorbored.VersionDef\x12\x13\n\x07version\x18\x03 \x01(\x05\x42\x02\x18\x01\x12\x30\n\x07library\x18\x02 \x01(\x0b\x32\x1f.tensorbored.FunctionDefLibrary\x12/\n\ndebug_info\x18\x05 \x01(\x0b\x32\x1b.tensorbored.GraphDebugInfoBz\n\x18org.tensorflow.frameworkB\x0bGraphProtosP\x01ZLgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/graph_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_GRAPHDEF = DESCRIPTOR.message_types_by_name['GraphDef']
+GraphDef = _reflection.GeneratedProtocolMessageType('GraphDef', (_message.Message,), {
+  'DESCRIPTOR' : _GRAPHDEF,
+  '__module__' : 'tensorbored.compat.proto.graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.GraphDef)
+  })
+_sym_db.RegisterMessage(GraphDef)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\013GraphProtosP\001ZLgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/graph_go_proto\370\001\001'
+  _GRAPHDEF.fields_by_name['version']._options = None
+  _GRAPHDEF.fields_by_name['version']._serialized_options = b'\030\001'
+  _GRAPHDEF._serialized_start=226
+  _GRAPHDEF._serialized_end=435
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/histogram_pb2.py

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/histogram.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/histogram.proto\x12\x0btensorbored\"\x87\x01\n\x0eHistogramProto\x12\x0b\n\x03min\x18\x01 \x01(\x01\x12\x0b\n\x03max\x18\x02 \x01(\x01\x12\x0b\n\x03num\x18\x03 \x01(\x01\x12\x0b\n\x03sum\x18\x04 \x01(\x01\x12\x13\n\x0bsum_squares\x18\x05 \x01(\x01\x12\x18\n\x0c\x62ucket_limit\x18\x06 \x03(\x01\x42\x02\x10\x01\x12\x12\n\x06\x62ucket\x18\x07 \x03(\x01\x42\x02\x10\x01\x42\\\n\x18org.tensorflow.frameworkP\x01Z;github.com/google/tsl/tsl/go/core/protobuf/summary_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_HISTOGRAMPROTO = DESCRIPTOR.message_types_by_name['HistogramProto']
+HistogramProto = _reflection.GeneratedProtocolMessageType('HistogramProto', (_message.Message,), {
+  'DESCRIPTOR' : _HISTOGRAMPROTO,
+  '__module__' : 'tensorbored.compat.proto.histogram_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.HistogramProto)
+  })
+_sym_db.RegisterMessage(HistogramProto)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkP\001Z;github.com/google/tsl/tsl/go/core/protobuf/summary_go_proto\370\001\001'
+  _HISTOGRAMPROTO.fields_by_name['bucket_limit']._options = None
+  _HISTOGRAMPROTO.fields_by_name['bucket_limit']._serialized_options = b'\020\001'
+  _HISTOGRAMPROTO.fields_by_name['bucket']._options = None
+  _HISTOGRAMPROTO.fields_by_name['bucket']._serialized_options = b'\020\001'
+  _HISTOGRAMPROTO._serialized_start=58
+  _HISTOGRAMPROTO._serialized_end=193
+# @@protoc_insertion_point(module_scope)