tensorbored 2.21.0rc1769983804__py3-none-any.whl

tensorbored/compat/proto/saved_object_graph_pb2.py

@@ -0,0 +1,193 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/saved_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 any_pb2 as google_dot_protobuf_dot_any__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
+from tensorbored.compat.proto import variable_pb2 as tensorbored_dot_compat_dot_proto_dot_variable__pb2
+from tensorbored.compat.proto import versions_pb2 as tensorbored_dot_compat_dot_proto_dot_versions__pb2
+from tensorbored.compat.proto import struct_pb2 as tensorbored_dot_compat_dot_proto_dot_struct__pb2
+from tensorbored.compat.proto import trackable_object_graph_pb2 as tensorbored_dot_compat_dot_proto_dot_trackable__object__graph__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1tensorbored/compat/proto/saved_object_graph.proto\x12\x0btensorbored\x1a\x19google/protobuf/any.proto\x1a+tensorbored/compat/proto/tensor_shape.proto\x1a$tensorbored/compat/proto/types.proto\x1a\'tensorbored/compat/proto/variable.proto\x1a\'tensorbored/compat/proto/versions.proto\x1a%tensorbored/compat/proto/struct.proto\x1a\x35tensorbored/compat/proto/trackable_object_graph.proto\"\xeb\x01\n\x10SavedObjectGraph\x12\'\n\x05nodes\x18\x01 \x03(\x0b\x32\x18.tensorbored.SavedObject\x12P\n\x12\x63oncrete_functions\x18\x02 \x03(\x0b\x32\x34.tensorbored.SavedObjectGraph.ConcreteFunctionsEntry\x1a\\\n\x16\x43oncreteFunctionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x31\n\x05value\x18\x02 \x01(\x0b\x32\".tensorbored.SavedConcreteFunction:\x02\x38\x01\"\xdd\x07\n\x0bSavedObject\x12S\n\x08\x63hildren\x18\x01 \x03(\x0b\x32\x41.tensorbored.TrackableObjectGraph.TrackableObject.ObjectReference\x12W\n\x0c\x64\x65pendencies\x18\x0f \x03(\x0b\x32\x41.tensorbored.TrackableObjectGraph.TrackableObject.ObjectReference\x12_\n\x0eslot_variables\x18\x03 \x03(\x0b\x32G.tensorbored.TrackableObjectGraph.TrackableObject.SlotVariableReference\x12\x33\n\x0buser_object\x18\x04 \x01(\x0b\x32\x1c.tensorbored.SavedUserObjectH\x00\x12(\n\x05\x61sset\x18\x05 \x01(\x0b\x32\x17.tensorbored.SavedAssetH\x00\x12.\n\x08\x66unction\x18\x06 \x01(\x0b\x32\x1a.tensorbored.SavedFunctionH\x00\x12.\n\x08variable\x18\x07 \x01(\x0b\x32\x1a.tensorbored.SavedVariableH\x00\x12H\n\x16\x62\x61re_concrete_function\x18\x08 \x01(\x0b\x32&.tensorbored.SavedBareConcreteFunctionH\x00\x12.\n\x08\x63onstant\x18\t \x01(\x0b\x32\x1a.tensorbored.SavedConstantH\x00\x12.\n\x08resource\x18\n \x01(\x0b\x32\x1a.tensorbored.SavedResourceH\x00\x12\x36\n\x0f\x63\x61ptured_tensor\x18\x0c \x01(\x0b\x32\x1b.tensorbored.CapturedTensorH\x00\x12G\n\x10saveable_objects\x18\x0b \x03(\x0b\x32-.tensorbored.SavedObject.SaveableObjectsEntry\x12\x17\n\x0fregistered_name\x18\r \x01(\t\x12\x33\n\x15serialized_user_proto\x18\x0e \x01(\x0b\x32\x14.google.protobuf.Any\x12\x18\n\x10registered_saver\x18\x10 \x01(\t\x1aS\n\x14SaveableObjectsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12*\n\x05value\x18\x02 \x01(\x0b\x32\x1b.tensorbored.SaveableObject:\x02\x38\x01\x42\x06\n\x04kindJ\x04\x08\x02\x10\x03R\nattributes\"e\n\x0fSavedUserObject\x12\x12\n\nidentifier\x18\x01 \x01(\t\x12(\n\x07version\x18\x02 \x01(\x0b\x32\x17.tensorbored.VersionDef\x12\x14\n\x08metadata\x18\x03 \x01(\tB\x02\x18\x01\"*\n\nSavedAsset\x12\x1c\n\x14\x61sset_file_def_index\x18\x01 \x01(\x05\"]\n\rSavedFunction\x12\x1a\n\x12\x63oncrete_functions\x18\x01 \x03(\t\x12\x30\n\rfunction_spec\x18\x02 \x01(\x0b\x32\x19.tensorbored.FunctionSpec\"9\n\x0e\x43\x61pturedTensor\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x19\n\x11\x63oncrete_function\x18\x02 \x01(\t\"\xaa\x01\n\x15SavedConcreteFunction\x12\x14\n\x0c\x62ound_inputs\x18\x02 \x03(\x05\x12\x43\n\x1d\x63\x61nonicalized_input_signature\x18\x03 \x01(\x0b\x32\x1c.tensorbored.StructuredValue\x12\x36\n\x10output_signature\x18\x04 \x01(\x0b\x32\x1c.tensorbored.StructuredValue\"\xae\x01\n\x19SavedBareConcreteFunction\x12\x1e\n\x16\x63oncrete_function_name\x18\x01 \x01(\t\x12\x19\n\x11\x61rgument_keywords\x18\x02 \x03(\t\x12$\n\x1c\x61llowed_positional_arguments\x18\x03 \x01(\x03\x12\x30\n\rfunction_spec\x18\x04 \x01(\x0b\x32\x19.tensorbored.FunctionSpec\"\"\n\rSavedConstant\x12\x11\n\toperation\x18\x01 \x01(\t\"\xdc\x02\n\rSavedVariable\x12$\n\x05\x64type\x18\x01 \x01(\x0e\x32\x15.tensorbored.DataType\x12,\n\x05shape\x18\x02 \x01(\x0b\x32\x1d.tensorbored.TensorShapeProto\x12\x11\n\ttrainable\x18\x03 \x01(\x08\x12=\n\x0fsynchronization\x18\x04 \x01(\x0e\x32$.tensorbored.VariableSynchronization\x12\x35\n\x0b\x61ggregation\x18\x05 \x01(\x0e\x32 .tensorbored.VariableAggregation\x12\x0c\n\x04name\x18\x06 \x01(\t\x12\x0e\n\x06\x64\x65vice\x18\x07 \x01(\t\x12P\n,experimental_distributed_variable_components\x18\x08 \x03(\x0b\x32\x1a.tensorbored.SavedVariable\"\xfe\x01\n\x0c\x46unctionSpec\x12\x31\n\x0b\x66ullargspec\x18\x01 \x01(\x0b\x32\x1c.tensorbored.StructuredValue\x12\x11\n\tis_method\x18\x02 \x01(\x08\x12\x35\n\x0finput_signature\x18\x05 \x01(\x0b\x32\x1c.tensorbored.StructuredValue\x12\x39\n\x0bjit_compile\x18\x06 \x01(\x0e\x32$.tensorbored.FunctionSpec.JitCompile\"*\n\nJitCompile\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x06\n\x02ON\x10\x01\x12\x07\n\x03OFF\x10\x02J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05\"\x1f\n\rSavedResource\x12\x0e\n\x06\x64\x65vice\x18\x01 \x01(\t\"A\n\x0eSaveableObject\x12\x15\n\rsave_function\x18\x02 \x01(\x05\x12\x18\n\x10restore_function\x18\x03 \x01(\x05\x42ZZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_SAVEDOBJECTGRAPH = DESCRIPTOR.message_types_by_name['SavedObjectGraph']
+_SAVEDOBJECTGRAPH_CONCRETEFUNCTIONSENTRY = _SAVEDOBJECTGRAPH.nested_types_by_name['ConcreteFunctionsEntry']
+_SAVEDOBJECT = DESCRIPTOR.message_types_by_name['SavedObject']
+_SAVEDOBJECT_SAVEABLEOBJECTSENTRY = _SAVEDOBJECT.nested_types_by_name['SaveableObjectsEntry']
+_SAVEDUSEROBJECT = DESCRIPTOR.message_types_by_name['SavedUserObject']
+_SAVEDASSET = DESCRIPTOR.message_types_by_name['SavedAsset']
+_SAVEDFUNCTION = DESCRIPTOR.message_types_by_name['SavedFunction']
+_CAPTUREDTENSOR = DESCRIPTOR.message_types_by_name['CapturedTensor']
+_SAVEDCONCRETEFUNCTION = DESCRIPTOR.message_types_by_name['SavedConcreteFunction']
+_SAVEDBARECONCRETEFUNCTION = DESCRIPTOR.message_types_by_name['SavedBareConcreteFunction']
+_SAVEDCONSTANT = DESCRIPTOR.message_types_by_name['SavedConstant']
+_SAVEDVARIABLE = DESCRIPTOR.message_types_by_name['SavedVariable']
+_FUNCTIONSPEC = DESCRIPTOR.message_types_by_name['FunctionSpec']
+_SAVEDRESOURCE = DESCRIPTOR.message_types_by_name['SavedResource']
+_SAVEABLEOBJECT = DESCRIPTOR.message_types_by_name['SaveableObject']
+_FUNCTIONSPEC_JITCOMPILE = _FUNCTIONSPEC.enum_types_by_name['JitCompile']
+SavedObjectGraph = _reflection.GeneratedProtocolMessageType('SavedObjectGraph', (_message.Message,), {
+
+  'ConcreteFunctionsEntry' : _reflection.GeneratedProtocolMessageType('ConcreteFunctionsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _SAVEDOBJECTGRAPH_CONCRETEFUNCTIONSENTRY,
+    '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.SavedObjectGraph.ConcreteFunctionsEntry)
+    })
+  ,
+  'DESCRIPTOR' : _SAVEDOBJECTGRAPH,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SavedObjectGraph)
+  })
+_sym_db.RegisterMessage(SavedObjectGraph)
+_sym_db.RegisterMessage(SavedObjectGraph.ConcreteFunctionsEntry)
+
+SavedObject = _reflection.GeneratedProtocolMessageType('SavedObject', (_message.Message,), {
+
+  'SaveableObjectsEntry' : _reflection.GeneratedProtocolMessageType('SaveableObjectsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _SAVEDOBJECT_SAVEABLEOBJECTSENTRY,
+    '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.SavedObject.SaveableObjectsEntry)
+    })
+  ,
+  'DESCRIPTOR' : _SAVEDOBJECT,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SavedObject)
+  })
+_sym_db.RegisterMessage(SavedObject)
+_sym_db.RegisterMessage(SavedObject.SaveableObjectsEntry)
+
+SavedUserObject = _reflection.GeneratedProtocolMessageType('SavedUserObject', (_message.Message,), {
+  'DESCRIPTOR' : _SAVEDUSEROBJECT,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SavedUserObject)
+  })
+_sym_db.RegisterMessage(SavedUserObject)
+
+SavedAsset = _reflection.GeneratedProtocolMessageType('SavedAsset', (_message.Message,), {
+  'DESCRIPTOR' : _SAVEDASSET,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SavedAsset)
+  })
+_sym_db.RegisterMessage(SavedAsset)
+
+SavedFunction = _reflection.GeneratedProtocolMessageType('SavedFunction', (_message.Message,), {
+  'DESCRIPTOR' : _SAVEDFUNCTION,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SavedFunction)
+  })
+_sym_db.RegisterMessage(SavedFunction)
+
+CapturedTensor = _reflection.GeneratedProtocolMessageType('CapturedTensor', (_message.Message,), {
+  'DESCRIPTOR' : _CAPTUREDTENSOR,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.CapturedTensor)
+  })
+_sym_db.RegisterMessage(CapturedTensor)
+
+SavedConcreteFunction = _reflection.GeneratedProtocolMessageType('SavedConcreteFunction', (_message.Message,), {
+  'DESCRIPTOR' : _SAVEDCONCRETEFUNCTION,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SavedConcreteFunction)
+  })
+_sym_db.RegisterMessage(SavedConcreteFunction)
+
+SavedBareConcreteFunction = _reflection.GeneratedProtocolMessageType('SavedBareConcreteFunction', (_message.Message,), {
+  'DESCRIPTOR' : _SAVEDBARECONCRETEFUNCTION,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SavedBareConcreteFunction)
+  })
+_sym_db.RegisterMessage(SavedBareConcreteFunction)
+
+SavedConstant = _reflection.GeneratedProtocolMessageType('SavedConstant', (_message.Message,), {
+  'DESCRIPTOR' : _SAVEDCONSTANT,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SavedConstant)
+  })
+_sym_db.RegisterMessage(SavedConstant)
+
+SavedVariable = _reflection.GeneratedProtocolMessageType('SavedVariable', (_message.Message,), {
+  'DESCRIPTOR' : _SAVEDVARIABLE,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SavedVariable)
+  })
+_sym_db.RegisterMessage(SavedVariable)
+
+FunctionSpec = _reflection.GeneratedProtocolMessageType('FunctionSpec', (_message.Message,), {
+  'DESCRIPTOR' : _FUNCTIONSPEC,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.FunctionSpec)
+  })
+_sym_db.RegisterMessage(FunctionSpec)
+
+SavedResource = _reflection.GeneratedProtocolMessageType('SavedResource', (_message.Message,), {
+  'DESCRIPTOR' : _SAVEDRESOURCE,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SavedResource)
+  })
+_sym_db.RegisterMessage(SavedResource)
+
+SaveableObject = _reflection.GeneratedProtocolMessageType('SaveableObject', (_message.Message,), {
+  'DESCRIPTOR' : _SAVEABLEOBJECT,
+  '__module__' : 'tensorbored.compat.proto.saved_object_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SaveableObject)
+  })
+_sym_db.RegisterMessage(SaveableObject)
+
+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'
+  _SAVEDOBJECTGRAPH_CONCRETEFUNCTIONSENTRY._options = None
+  _SAVEDOBJECTGRAPH_CONCRETEFUNCTIONSENTRY._serialized_options = b'8\001'
+  _SAVEDOBJECT_SAVEABLEOBJECTSENTRY._options = None
+  _SAVEDOBJECT_SAVEABLEOBJECTSENTRY._serialized_options = b'8\001'
+  _SAVEDUSEROBJECT.fields_by_name['metadata']._options = None
+  _SAVEDUSEROBJECT.fields_by_name['metadata']._serialized_options = b'\030\001'
+  _SAVEDOBJECTGRAPH._serialized_start=353
+  _SAVEDOBJECTGRAPH._serialized_end=588
+  _SAVEDOBJECTGRAPH_CONCRETEFUNCTIONSENTRY._serialized_start=496
+  _SAVEDOBJECTGRAPH_CONCRETEFUNCTIONSENTRY._serialized_end=588
+  _SAVEDOBJECT._serialized_start=591
+  _SAVEDOBJECT._serialized_end=1580
+  _SAVEDOBJECT_SAVEABLEOBJECTSENTRY._serialized_start=1471
+  _SAVEDOBJECT_SAVEABLEOBJECTSENTRY._serialized_end=1554
+  _SAVEDUSEROBJECT._serialized_start=1582
+  _SAVEDUSEROBJECT._serialized_end=1683
+  _SAVEDASSET._serialized_start=1685
+  _SAVEDASSET._serialized_end=1727
+  _SAVEDFUNCTION._serialized_start=1729
+  _SAVEDFUNCTION._serialized_end=1822
+  _CAPTUREDTENSOR._serialized_start=1824
+  _CAPTUREDTENSOR._serialized_end=1881
+  _SAVEDCONCRETEFUNCTION._serialized_start=1884
+  _SAVEDCONCRETEFUNCTION._serialized_end=2054
+  _SAVEDBARECONCRETEFUNCTION._serialized_start=2057
+  _SAVEDBARECONCRETEFUNCTION._serialized_end=2231
+  _SAVEDCONSTANT._serialized_start=2233
+  _SAVEDCONSTANT._serialized_end=2267
+  _SAVEDVARIABLE._serialized_start=2270
+  _SAVEDVARIABLE._serialized_end=2618
+  _FUNCTIONSPEC._serialized_start=2621
+  _FUNCTIONSPEC._serialized_end=2875
+  _FUNCTIONSPEC_JITCOMPILE._serialized_start=2821
+  _FUNCTIONSPEC_JITCOMPILE._serialized_end=2863
+  _SAVEDRESOURCE._serialized_start=2877
+  _SAVEDRESOURCE._serialized_end=2908
+  _SAVEABLEOBJECT._serialized_start=2910
+  _SAVEABLEOBJECT._serialized_end=2975
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/saver_pb2.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/saver.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/saver.proto\x12\x0btensorbored\"\x9f\x02\n\x08SaverDef\x12\x1c\n\x14\x66ilename_tensor_name\x18\x01 \x01(\t\x12\x18\n\x10save_tensor_name\x18\x02 \x01(\t\x12\x17\n\x0frestore_op_name\x18\x03 \x01(\t\x12\x13\n\x0bmax_to_keep\x18\x04 \x01(\x05\x12\x0f\n\x07sharded\x18\x05 \x01(\x08\x12%\n\x1dkeep_checkpoint_every_n_hours\x18\x06 \x01(\x02\x12>\n\x07version\x18\x07 \x01(\x0e\x32-.tensorbored.SaverDef.CheckpointFormatVersion\"5\n\x17\x43heckpointFormatVersion\x12\n\n\x06LEGACY\x10\x00\x12\x06\n\x02V1\x10\x01\x12\x06\n\x02V2\x10\x02\x42~\n\x13org.tensorflow.utilB\x0bSaverProtosP\x01ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_SAVERDEF = DESCRIPTOR.message_types_by_name['SaverDef']
+_SAVERDEF_CHECKPOINTFORMATVERSION = _SAVERDEF.enum_types_by_name['CheckpointFormatVersion']
+SaverDef = _reflection.GeneratedProtocolMessageType('SaverDef', (_message.Message,), {
+  'DESCRIPTOR' : _SAVERDEF,
+  '__module__' : 'tensorbored.compat.proto.saver_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SaverDef)
+  })
+_sym_db.RegisterMessage(SaverDef)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\023org.tensorflow.utilB\013SaverProtosP\001ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\370\001\001'
+  _SAVERDEF._serialized_start=54
+  _SAVERDEF._serialized_end=341
+  _SAVERDEF_CHECKPOINTFORMATVERSION._serialized_start=288
+  _SAVERDEF_CHECKPOINTFORMATVERSION._serialized_end=341
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/step_stats_pb2.py

@@ -0,0 +1,116 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/step_stats.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_description_pb2 as tensorbored_dot_compat_dot_proto_dot_tensor__description__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)tensorbored/compat/proto/step_stats.proto\x12\x0btensorbored\x1a\x35tensorbored/compat/proto/allocation_description.proto\x1a\x31tensorbored/compat/proto/tensor_description.proto\"=\n\x10\x41llocationRecord\x12\x14\n\x0c\x61lloc_micros\x18\x01 \x01(\x03\x12\x13\n\x0b\x61lloc_bytes\x18\x02 \x01(\x03\"\xc5\x01\n\x13\x41llocatorMemoryUsed\x12\x16\n\x0e\x61llocator_name\x18\x01 \x01(\t\x12\x13\n\x0btotal_bytes\x18\x02 \x01(\x03\x12\x12\n\npeak_bytes\x18\x03 \x01(\x03\x12\x12\n\nlive_bytes\x18\x04 \x01(\x03\x12\x39\n\x12\x61llocation_records\x18\x06 \x03(\x0b\x32\x1d.tensorbored.AllocationRecord\x12\x1e\n\x16\x61llocator_bytes_in_use\x18\x05 \x01(\x03\"V\n\nNodeOutput\x12\x0c\n\x04slot\x18\x01 \x01(\x05\x12:\n\x12tensor_description\x18\x03 \x01(\x0b\x32\x1e.tensorbored.TensorDescription\"\xec\x01\n\x0bMemoryStats\x12\x18\n\x10temp_memory_size\x18\x01 \x01(\x03\x12\x1e\n\x16persistent_memory_size\x18\x03 \x01(\x03\x12#\n\x1bpersistent_tensor_alloc_ids\x18\x05 \x03(\x03\x12#\n\x17\x64\x65vice_temp_memory_size\x18\x02 \x01(\x03\x42\x02\x18\x01\x12)\n\x1d\x64\x65vice_persistent_memory_size\x18\x04 \x01(\x03\x42\x02\x18\x01\x12.\n\"device_persistent_tensor_alloc_ids\x18\x06 \x03(\x03\x42\x02\x18\x01\"\xa2\x04\n\rNodeExecStats\x12\x11\n\tnode_name\x18\x01 \x01(\t\x12\x18\n\x10\x61ll_start_micros\x18\x02 \x01(\x03\x12\x1b\n\x13op_start_rel_micros\x18\x03 \x01(\x03\x12\x19\n\x11op_end_rel_micros\x18\x04 \x01(\x03\x12\x1a\n\x12\x61ll_end_rel_micros\x18\x05 \x01(\x03\x12\x30\n\x06memory\x18\x06 \x03(\x0b\x32 .tensorbored.AllocatorMemoryUsed\x12\'\n\x06output\x18\x07 \x03(\x0b\x32\x17.tensorbored.NodeOutput\x12\x16\n\x0etimeline_label\x18\x08 \x01(\t\x12\x18\n\x10scheduled_micros\x18\t \x01(\x03\x12\x11\n\tthread_id\x18\n \x01(\r\x12=\n\x11referenced_tensor\x18\x0b \x03(\x0b\x32\".tensorbored.AllocationDescription\x12.\n\x0cmemory_stats\x18\x0c \x01(\x0b\x32\x18.tensorbored.MemoryStats\x12\x17\n\x0f\x61ll_start_nanos\x18\r \x01(\x03\x12\x1a\n\x12op_start_rel_nanos\x18\x0e \x01(\x03\x12\x18\n\x10op_end_rel_nanos\x18\x0f \x01(\x03\x12\x19\n\x11\x61ll_end_rel_nanos\x18\x10 \x01(\x03\x12\x17\n\x0fscheduled_nanos\x18\x11 \x01(\x03\"\xca\x01\n\x0f\x44\x65viceStepStats\x12\x0e\n\x06\x64\x65vice\x18\x01 \x01(\t\x12.\n\nnode_stats\x18\x02 \x03(\x0b\x32\x1a.tensorbored.NodeExecStats\x12\x43\n\x0cthread_names\x18\x03 \x03(\x0b\x32-.tensorbored.DeviceStepStats.ThreadNamesEntry\x1a\x32\n\x10ThreadNamesEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"<\n\tStepStats\x12/\n\tdev_stats\x18\x01 \x03(\x0b\x32\x1c.tensorbored.DeviceStepStatsB\x83\x01\n\x18org.tensorflow.frameworkB\x0fStepStatsProtosP\x01ZQgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/step_stats_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_ALLOCATIONRECORD = DESCRIPTOR.message_types_by_name['AllocationRecord']
+_ALLOCATORMEMORYUSED = DESCRIPTOR.message_types_by_name['AllocatorMemoryUsed']
+_NODEOUTPUT = DESCRIPTOR.message_types_by_name['NodeOutput']
+_MEMORYSTATS = DESCRIPTOR.message_types_by_name['MemoryStats']
+_NODEEXECSTATS = DESCRIPTOR.message_types_by_name['NodeExecStats']
+_DEVICESTEPSTATS = DESCRIPTOR.message_types_by_name['DeviceStepStats']
+_DEVICESTEPSTATS_THREADNAMESENTRY = _DEVICESTEPSTATS.nested_types_by_name['ThreadNamesEntry']
+_STEPSTATS = DESCRIPTOR.message_types_by_name['StepStats']
+AllocationRecord = _reflection.GeneratedProtocolMessageType('AllocationRecord', (_message.Message,), {
+  'DESCRIPTOR' : _ALLOCATIONRECORD,
+  '__module__' : 'tensorbored.compat.proto.step_stats_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.AllocationRecord)
+  })
+_sym_db.RegisterMessage(AllocationRecord)
+
+AllocatorMemoryUsed = _reflection.GeneratedProtocolMessageType('AllocatorMemoryUsed', (_message.Message,), {
+  'DESCRIPTOR' : _ALLOCATORMEMORYUSED,
+  '__module__' : 'tensorbored.compat.proto.step_stats_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.AllocatorMemoryUsed)
+  })
+_sym_db.RegisterMessage(AllocatorMemoryUsed)
+
+NodeOutput = _reflection.GeneratedProtocolMessageType('NodeOutput', (_message.Message,), {
+  'DESCRIPTOR' : _NODEOUTPUT,
+  '__module__' : 'tensorbored.compat.proto.step_stats_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.NodeOutput)
+  })
+_sym_db.RegisterMessage(NodeOutput)
+
+MemoryStats = _reflection.GeneratedProtocolMessageType('MemoryStats', (_message.Message,), {
+  'DESCRIPTOR' : _MEMORYSTATS,
+  '__module__' : 'tensorbored.compat.proto.step_stats_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.MemoryStats)
+  })
+_sym_db.RegisterMessage(MemoryStats)
+
+NodeExecStats = _reflection.GeneratedProtocolMessageType('NodeExecStats', (_message.Message,), {
+  'DESCRIPTOR' : _NODEEXECSTATS,
+  '__module__' : 'tensorbored.compat.proto.step_stats_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.NodeExecStats)
+  })
+_sym_db.RegisterMessage(NodeExecStats)
+
+DeviceStepStats = _reflection.GeneratedProtocolMessageType('DeviceStepStats', (_message.Message,), {
+
+  'ThreadNamesEntry' : _reflection.GeneratedProtocolMessageType('ThreadNamesEntry', (_message.Message,), {
+    'DESCRIPTOR' : _DEVICESTEPSTATS_THREADNAMESENTRY,
+    '__module__' : 'tensorbored.compat.proto.step_stats_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.DeviceStepStats.ThreadNamesEntry)
+    })
+  ,
+  'DESCRIPTOR' : _DEVICESTEPSTATS,
+  '__module__' : 'tensorbored.compat.proto.step_stats_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.DeviceStepStats)
+  })
+_sym_db.RegisterMessage(DeviceStepStats)
+_sym_db.RegisterMessage(DeviceStepStats.ThreadNamesEntry)
+
+StepStats = _reflection.GeneratedProtocolMessageType('StepStats', (_message.Message,), {
+  'DESCRIPTOR' : _STEPSTATS,
+  '__module__' : 'tensorbored.compat.proto.step_stats_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.StepStats)
+  })
+_sym_db.RegisterMessage(StepStats)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\017StepStatsProtosP\001ZQgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/step_stats_go_proto\370\001\001'
+  _MEMORYSTATS.fields_by_name['device_temp_memory_size']._options = None
+  _MEMORYSTATS.fields_by_name['device_temp_memory_size']._serialized_options = b'\030\001'
+  _MEMORYSTATS.fields_by_name['device_persistent_memory_size']._options = None
+  _MEMORYSTATS.fields_by_name['device_persistent_memory_size']._serialized_options = b'\030\001'
+  _MEMORYSTATS.fields_by_name['device_persistent_tensor_alloc_ids']._options = None
+  _MEMORYSTATS.fields_by_name['device_persistent_tensor_alloc_ids']._serialized_options = b'\030\001'
+  _DEVICESTEPSTATS_THREADNAMESENTRY._options = None
+  _DEVICESTEPSTATS_THREADNAMESENTRY._serialized_options = b'8\001'
+  _ALLOCATIONRECORD._serialized_start=164
+  _ALLOCATIONRECORD._serialized_end=225
+  _ALLOCATORMEMORYUSED._serialized_start=228
+  _ALLOCATORMEMORYUSED._serialized_end=425
+  _NODEOUTPUT._serialized_start=427
+  _NODEOUTPUT._serialized_end=513
+  _MEMORYSTATS._serialized_start=516
+  _MEMORYSTATS._serialized_end=752
+  _NODEEXECSTATS._serialized_start=755
+  _NODEEXECSTATS._serialized_end=1301
+  _DEVICESTEPSTATS._serialized_start=1304
+  _DEVICESTEPSTATS._serialized_end=1506
+  _DEVICESTEPSTATS_THREADNAMESENTRY._serialized_start=1456
+  _DEVICESTEPSTATS_THREADNAMESENTRY._serialized_end=1506
+  _STEPSTATS._serialized_start=1508
+  _STEPSTATS._serialized_end=1568
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/struct_pb2.py

@@ -0,0 +1,144 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/struct.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 tensor_pb2 as tensorbored_dot_compat_dot_proto_dot_tensor__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/struct.proto\x12\x0btensorbored\x1a%tensorbored/compat/proto/tensor.proto\x1a+tensorbored/compat/proto/tensor_shape.proto\x1a$tensorbored/compat/proto/types.proto\"\xfd\x05\n\x0fStructuredValue\x12,\n\nnone_value\x18\x01 \x01(\x0b\x32\x16.tensorbored.NoneValueH\x00\x12\x17\n\rfloat64_value\x18\x0b \x01(\x01H\x00\x12\x15\n\x0bint64_value\x18\x0c \x01(\x12H\x00\x12\x16\n\x0cstring_value\x18\r \x01(\tH\x00\x12\x14\n\nbool_value\x18\x0e \x01(\x08H\x00\x12;\n\x12tensor_shape_value\x18\x1f \x01(\x0b\x32\x1d.tensorbored.TensorShapeProtoH\x00\x12\x33\n\x12tensor_dtype_value\x18  \x01(\x0e\x32\x15.tensorbored.DataTypeH\x00\x12\x39\n\x11tensor_spec_value\x18! \x01(\x0b\x32\x1c.tensorbored.TensorSpecProtoH\x00\x12\x35\n\x0ftype_spec_value\x18\" \x01(\x0b\x32\x1a.tensorbored.TypeSpecProtoH\x00\x12H\n\x19\x62ounded_tensor_spec_value\x18# \x01(\x0b\x32#.tensorbored.BoundedTensorSpecProtoH\x00\x12,\n\nlist_value\x18\x33 \x01(\x0b\x32\x16.tensorbored.ListValueH\x00\x12.\n\x0btuple_value\x18\x34 \x01(\x0b\x32\x17.tensorbored.TupleValueH\x00\x12,\n\ndict_value\x18\x35 \x01(\x0b\x32\x16.tensorbored.DictValueH\x00\x12\x39\n\x11named_tuple_value\x18\x36 \x01(\x0b\x32\x1c.tensorbored.NamedTupleValueH\x00\x12\x30\n\x0ctensor_value\x18\x37 \x01(\x0b\x32\x18.tensorbored.TensorProtoH\x00\x12/\n\x0bnumpy_value\x18\x38 \x01(\x0b\x32\x18.tensorbored.TensorProtoH\x00\x42\x06\n\x04kind\"\x0b\n\tNoneValue\"9\n\tListValue\x12,\n\x06values\x18\x01 \x03(\x0b\x32\x1c.tensorbored.StructuredValue\":\n\nTupleValue\x12,\n\x06values\x18\x01 \x03(\x0b\x32\x1c.tensorbored.StructuredValue\"\x8c\x01\n\tDictValue\x12\x32\n\x06\x66ields\x18\x01 \x03(\x0b\x32\".tensorbored.DictValue.FieldsEntry\x1aK\n\x0b\x46ieldsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12+\n\x05value\x18\x02 \x01(\x0b\x32\x1c.tensorbored.StructuredValue:\x02\x38\x01\"E\n\tPairValue\x12\x0b\n\x03key\x18\x01 \x01(\t\x12+\n\x05value\x18\x02 \x01(\x0b\x32\x1c.tensorbored.StructuredValue\"G\n\x0fNamedTupleValue\x12\x0c\n\x04name\x18\x01 \x01(\t\x12&\n\x06values\x18\x02 \x03(\x0b\x32\x16.tensorbored.PairValue\"s\n\x0fTensorSpecProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12,\n\x05shape\x18\x02 \x01(\x0b\x32\x1d.tensorbored.TensorShapeProto\x12$\n\x05\x64type\x18\x03 \x01(\x0e\x32\x15.tensorbored.DataType\"\xd0\x01\n\x16\x42oundedTensorSpecProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12,\n\x05shape\x18\x02 \x01(\x0b\x32\x1d.tensorbored.TensorShapeProto\x12$\n\x05\x64type\x18\x03 \x01(\x0e\x32\x15.tensorbored.DataType\x12)\n\x07minimum\x18\x04 \x01(\x0b\x32\x18.tensorbored.TensorProto\x12)\n\x07maximum\x18\x05 \x01(\x0b\x32\x18.tensorbored.TensorProto\"\xfa\x03\n\rTypeSpecProto\x12\x41\n\x0ftype_spec_class\x18\x01 \x01(\x0e\x32(.tensorbored.TypeSpecProto.TypeSpecClass\x12\x30\n\ntype_state\x18\x02 \x01(\x0b\x32\x1c.tensorbored.StructuredValue\x12\x1c\n\x14type_spec_class_name\x18\x03 \x01(\t\x12\x1b\n\x13num_flat_components\x18\x04 \x01(\x05\"\xb8\x02\n\rTypeSpecClass\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x16\n\x12SPARSE_TENSOR_SPEC\x10\x01\x12\x17\n\x13INDEXED_SLICES_SPEC\x10\x02\x12\x16\n\x12RAGGED_TENSOR_SPEC\x10\x03\x12\x15\n\x11TENSOR_ARRAY_SPEC\x10\x04\x12\x15\n\x11\x44\x41TA_DATASET_SPEC\x10\x05\x12\x16\n\x12\x44\x41TA_ITERATOR_SPEC\x10\x06\x12\x11\n\rOPTIONAL_SPEC\x10\x07\x12\x14\n\x10PER_REPLICA_SPEC\x10\x08\x12\x11\n\rVARIABLE_SPEC\x10\t\x12\x16\n\x12ROW_PARTITION_SPEC\x10\n\x12\x18\n\x14REGISTERED_TYPE_SPEC\x10\x0c\x12\x17\n\x13\x45XTENSION_TYPE_SPEC\x10\r\"\x04\x08\x0b\x10\x0b\x42WZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_protob\x06proto3')
+
+
+
+_STRUCTUREDVALUE = DESCRIPTOR.message_types_by_name['StructuredValue']
+_NONEVALUE = DESCRIPTOR.message_types_by_name['NoneValue']
+_LISTVALUE = DESCRIPTOR.message_types_by_name['ListValue']
+_TUPLEVALUE = DESCRIPTOR.message_types_by_name['TupleValue']
+_DICTVALUE = DESCRIPTOR.message_types_by_name['DictValue']
+_DICTVALUE_FIELDSENTRY = _DICTVALUE.nested_types_by_name['FieldsEntry']
+_PAIRVALUE = DESCRIPTOR.message_types_by_name['PairValue']
+_NAMEDTUPLEVALUE = DESCRIPTOR.message_types_by_name['NamedTupleValue']
+_TENSORSPECPROTO = DESCRIPTOR.message_types_by_name['TensorSpecProto']
+_BOUNDEDTENSORSPECPROTO = DESCRIPTOR.message_types_by_name['BoundedTensorSpecProto']
+_TYPESPECPROTO = DESCRIPTOR.message_types_by_name['TypeSpecProto']
+_TYPESPECPROTO_TYPESPECCLASS = _TYPESPECPROTO.enum_types_by_name['TypeSpecClass']
+StructuredValue = _reflection.GeneratedProtocolMessageType('StructuredValue', (_message.Message,), {
+  'DESCRIPTOR' : _STRUCTUREDVALUE,
+  '__module__' : 'tensorbored.compat.proto.struct_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.StructuredValue)
+  })
+_sym_db.RegisterMessage(StructuredValue)
+
+NoneValue = _reflection.GeneratedProtocolMessageType('NoneValue', (_message.Message,), {
+  'DESCRIPTOR' : _NONEVALUE,
+  '__module__' : 'tensorbored.compat.proto.struct_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.NoneValue)
+  })
+_sym_db.RegisterMessage(NoneValue)
+
+ListValue = _reflection.GeneratedProtocolMessageType('ListValue', (_message.Message,), {
+  'DESCRIPTOR' : _LISTVALUE,
+  '__module__' : 'tensorbored.compat.proto.struct_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ListValue)
+  })
+_sym_db.RegisterMessage(ListValue)
+
+TupleValue = _reflection.GeneratedProtocolMessageType('TupleValue', (_message.Message,), {
+  'DESCRIPTOR' : _TUPLEVALUE,
+  '__module__' : 'tensorbored.compat.proto.struct_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.TupleValue)
+  })
+_sym_db.RegisterMessage(TupleValue)
+
+DictValue = _reflection.GeneratedProtocolMessageType('DictValue', (_message.Message,), {
+
+  'FieldsEntry' : _reflection.GeneratedProtocolMessageType('FieldsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _DICTVALUE_FIELDSENTRY,
+    '__module__' : 'tensorbored.compat.proto.struct_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.DictValue.FieldsEntry)
+    })
+  ,
+  'DESCRIPTOR' : _DICTVALUE,
+  '__module__' : 'tensorbored.compat.proto.struct_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.DictValue)
+  })
+_sym_db.RegisterMessage(DictValue)
+_sym_db.RegisterMessage(DictValue.FieldsEntry)
+
+PairValue = _reflection.GeneratedProtocolMessageType('PairValue', (_message.Message,), {
+  'DESCRIPTOR' : _PAIRVALUE,
+  '__module__' : 'tensorbored.compat.proto.struct_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.PairValue)
+  })
+_sym_db.RegisterMessage(PairValue)
+
+NamedTupleValue = _reflection.GeneratedProtocolMessageType('NamedTupleValue', (_message.Message,), {
+  'DESCRIPTOR' : _NAMEDTUPLEVALUE,
+  '__module__' : 'tensorbored.compat.proto.struct_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.NamedTupleValue)
+  })
+_sym_db.RegisterMessage(NamedTupleValue)
+
+TensorSpecProto = _reflection.GeneratedProtocolMessageType('TensorSpecProto', (_message.Message,), {
+  'DESCRIPTOR' : _TENSORSPECPROTO,
+  '__module__' : 'tensorbored.compat.proto.struct_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.TensorSpecProto)
+  })
+_sym_db.RegisterMessage(TensorSpecProto)
+
+BoundedTensorSpecProto = _reflection.GeneratedProtocolMessageType('BoundedTensorSpecProto', (_message.Message,), {
+  'DESCRIPTOR' : _BOUNDEDTENSORSPECPROTO,
+  '__module__' : 'tensorbored.compat.proto.struct_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.BoundedTensorSpecProto)
+  })
+_sym_db.RegisterMessage(BoundedTensorSpecProto)
+
+TypeSpecProto = _reflection.GeneratedProtocolMessageType('TypeSpecProto', (_message.Message,), {
+  'DESCRIPTOR' : _TYPESPECPROTO,
+  '__module__' : 'tensorbored.compat.proto.struct_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.TypeSpecProto)
+  })
+_sym_db.RegisterMessage(TypeSpecProto)
+
+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'
+  _DICTVALUE_FIELDSENTRY._options = None
+  _DICTVALUE_FIELDSENTRY._serialized_options = b'8\001'
+  _STRUCTUREDVALUE._serialized_start=177
+  _STRUCTUREDVALUE._serialized_end=942
+  _NONEVALUE._serialized_start=944
+  _NONEVALUE._serialized_end=955
+  _LISTVALUE._serialized_start=957
+  _LISTVALUE._serialized_end=1014
+  _TUPLEVALUE._serialized_start=1016
+  _TUPLEVALUE._serialized_end=1074
+  _DICTVALUE._serialized_start=1077
+  _DICTVALUE._serialized_end=1217
+  _DICTVALUE_FIELDSENTRY._serialized_start=1142
+  _DICTVALUE_FIELDSENTRY._serialized_end=1217
+  _PAIRVALUE._serialized_start=1219
+  _PAIRVALUE._serialized_end=1288
+  _NAMEDTUPLEVALUE._serialized_start=1290
+  _NAMEDTUPLEVALUE._serialized_end=1361
+  _TENSORSPECPROTO._serialized_start=1363
+  _TENSORSPECPROTO._serialized_end=1478
+  _BOUNDEDTENSORSPECPROTO._serialized_start=1481
+  _BOUNDEDTENSORSPECPROTO._serialized_end=1689
+  _TYPESPECPROTO._serialized_start=1692
+  _TYPESPECPROTO._serialized_end=2198
+  _TYPESPECPROTO_TYPESPECCLASS._serialized_start=1886
+  _TYPESPECPROTO_TYPESPECCLASS._serialized_end=2198
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/summary_pb2.py

@@ -0,0 +1,111 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/summary.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 histogram_pb2 as tensorbored_dot_compat_dot_proto_dot_histogram__pb2
+from tensorbored.compat.proto import tensor_pb2 as tensorbored_dot_compat_dot_proto_dot_tensor__pb2
+
+from tensorbored.compat.proto.histogram_pb2 import *
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n&tensorbored/compat/proto/summary.proto\x12\x0btensorbored\x1a(tensorbored/compat/proto/histogram.proto\x1a%tensorbored/compat/proto/tensor.proto\"\'\n\x12SummaryDescription\x12\x11\n\ttype_hint\x18\x01 \x01(\t\"\xe2\x01\n\x0fSummaryMetadata\x12<\n\x0bplugin_data\x18\x01 \x01(\x0b\x32\'.tensorbored.SummaryMetadata.PluginData\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x1b\n\x13summary_description\x18\x03 \x01(\t\x12*\n\ndata_class\x18\x04 \x01(\x0e\x32\x16.tensorbored.DataClass\x1a\x32\n\nPluginData\x12\x13\n\x0bplugin_name\x18\x01 \x01(\t\x12\x0f\n\x07\x63ontent\x18\x02 \x01(\x0c\"\xe4\x04\n\x07Summary\x12)\n\x05value\x18\x01 \x03(\x0b\x32\x1a.tensorbored.Summary.Value\x1aX\n\x05Image\x12\x0e\n\x06height\x18\x01 \x01(\x05\x12\r\n\x05width\x18\x02 \x01(\x05\x12\x12\n\ncolorspace\x18\x03 \x01(\x05\x12\x1c\n\x14\x65ncoded_image_string\x18\x04 \x01(\x0c\x1a}\n\x05\x41udio\x12\x13\n\x0bsample_rate\x18\x01 \x01(\x02\x12\x14\n\x0cnum_channels\x18\x02 \x01(\x03\x12\x15\n\rlength_frames\x18\x03 \x01(\x03\x12\x1c\n\x14\x65ncoded_audio_string\x18\x04 \x01(\x0c\x12\x14\n\x0c\x63ontent_type\x18\x05 \x01(\t\x1a\xd4\x02\n\x05Value\x12\x11\n\tnode_name\x18\x07 \x01(\t\x12\x0b\n\x03tag\x18\x01 \x01(\t\x12.\n\x08metadata\x18\t \x01(\x0b\x32\x1c.tensorbored.SummaryMetadata\x12\x16\n\x0csimple_value\x18\x02 \x01(\x02H\x00\x12&\n\x1cobsolete_old_style_histogram\x18\x03 \x01(\x0cH\x00\x12+\n\x05image\x18\x04 \x01(\x0b\x32\x1a.tensorbored.Summary.ImageH\x00\x12,\n\x05histo\x18\x05 \x01(\x0b\x32\x1b.tensorbored.HistogramProtoH\x00\x12+\n\x05\x61udio\x18\x06 \x01(\x0b\x32\x1a.tensorbored.Summary.AudioH\x00\x12*\n\x06tensor\x18\x08 \x01(\x0b\x32\x18.tensorbored.TensorProtoH\x00\x42\x07\n\x05value*o\n\tDataClass\x12\x16\n\x12\x44\x41TA_CLASS_UNKNOWN\x10\x00\x12\x15\n\x11\x44\x41TA_CLASS_SCALAR\x10\x01\x12\x15\n\x11\x44\x41TA_CLASS_TENSOR\x10\x02\x12\x1c\n\x18\x44\x41TA_CLASS_BLOB_SEQUENCE\x10\x03\x42~\n\x18org.tensorflow.frameworkB\rSummaryProtosP\x01ZNgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/summary_go_proto\xf8\x01\x01P\x00\x62\x06proto3')
+
+_DATACLASS = DESCRIPTOR.enum_types_by_name['DataClass']
+DataClass = enum_type_wrapper.EnumTypeWrapper(_DATACLASS)
+DATA_CLASS_UNKNOWN = 0
+DATA_CLASS_SCALAR = 1
+DATA_CLASS_TENSOR = 2
+DATA_CLASS_BLOB_SEQUENCE = 3
+
+
+_SUMMARYDESCRIPTION = DESCRIPTOR.message_types_by_name['SummaryDescription']
+_SUMMARYMETADATA = DESCRIPTOR.message_types_by_name['SummaryMetadata']
+_SUMMARYMETADATA_PLUGINDATA = _SUMMARYMETADATA.nested_types_by_name['PluginData']
+_SUMMARY = DESCRIPTOR.message_types_by_name['Summary']
+_SUMMARY_IMAGE = _SUMMARY.nested_types_by_name['Image']
+_SUMMARY_AUDIO = _SUMMARY.nested_types_by_name['Audio']
+_SUMMARY_VALUE = _SUMMARY.nested_types_by_name['Value']
+SummaryDescription = _reflection.GeneratedProtocolMessageType('SummaryDescription', (_message.Message,), {
+  'DESCRIPTOR' : _SUMMARYDESCRIPTION,
+  '__module__' : 'tensorbored.compat.proto.summary_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SummaryDescription)
+  })
+_sym_db.RegisterMessage(SummaryDescription)
+
+SummaryMetadata = _reflection.GeneratedProtocolMessageType('SummaryMetadata', (_message.Message,), {
+
+  'PluginData' : _reflection.GeneratedProtocolMessageType('PluginData', (_message.Message,), {
+    'DESCRIPTOR' : _SUMMARYMETADATA_PLUGINDATA,
+    '__module__' : 'tensorbored.compat.proto.summary_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.SummaryMetadata.PluginData)
+    })
+  ,
+  'DESCRIPTOR' : _SUMMARYMETADATA,
+  '__module__' : 'tensorbored.compat.proto.summary_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SummaryMetadata)
+  })
+_sym_db.RegisterMessage(SummaryMetadata)
+_sym_db.RegisterMessage(SummaryMetadata.PluginData)
+
+Summary = _reflection.GeneratedProtocolMessageType('Summary', (_message.Message,), {
+
+  'Image' : _reflection.GeneratedProtocolMessageType('Image', (_message.Message,), {
+    'DESCRIPTOR' : _SUMMARY_IMAGE,
+    '__module__' : 'tensorbored.compat.proto.summary_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.Summary.Image)
+    })
+  ,
+
+  'Audio' : _reflection.GeneratedProtocolMessageType('Audio', (_message.Message,), {
+    'DESCRIPTOR' : _SUMMARY_AUDIO,
+    '__module__' : 'tensorbored.compat.proto.summary_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.Summary.Audio)
+    })
+  ,
+
+  'Value' : _reflection.GeneratedProtocolMessageType('Value', (_message.Message,), {
+    'DESCRIPTOR' : _SUMMARY_VALUE,
+    '__module__' : 'tensorbored.compat.proto.summary_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.Summary.Value)
+    })
+  ,
+  'DESCRIPTOR' : _SUMMARY,
+  '__module__' : 'tensorbored.compat.proto.summary_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.Summary)
+  })
+_sym_db.RegisterMessage(Summary)
+_sym_db.RegisterMessage(Summary.Image)
+_sym_db.RegisterMessage(Summary.Audio)
+_sym_db.RegisterMessage(Summary.Value)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\rSummaryProtosP\001ZNgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/summary_go_proto\370\001\001'
+  _DATACLASS._serialized_start=1021
+  _DATACLASS._serialized_end=1132
+  _SUMMARYDESCRIPTION._serialized_start=136
+  _SUMMARYDESCRIPTION._serialized_end=175
+  _SUMMARYMETADATA._serialized_start=178
+  _SUMMARYMETADATA._serialized_end=404
+  _SUMMARYMETADATA_PLUGINDATA._serialized_start=354
+  _SUMMARYMETADATA_PLUGINDATA._serialized_end=404
+  _SUMMARY._serialized_start=407
+  _SUMMARY._serialized_end=1019
+  _SUMMARY_IMAGE._serialized_start=461
+  _SUMMARY_IMAGE._serialized_end=549
+  _SUMMARY_AUDIO._serialized_start=551
+  _SUMMARY_AUDIO._serialized_end=676
+  _SUMMARY_VALUE._serialized_start=679
+  _SUMMARY_VALUE._serialized_end=1019
+# @@protoc_insertion_point(module_scope)