tensorbored 2.21.0rc1769983804__py3-none-any.whl

tensorbored/compat/proto/meta_graph_pb2.py

@@ -0,0 +1,254 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/meta_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 graph_pb2 as tensorbored_dot_compat_dot_proto_dot_graph__pb2
+from tensorbored.compat.proto import op_def_pb2 as tensorbored_dot_compat_dot_proto_dot_op__def__pb2
+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
+from tensorbored.compat.proto import saved_object_graph_pb2 as tensorbored_dot_compat_dot_proto_dot_saved__object__graph__pb2
+from tensorbored.compat.proto import saver_pb2 as tensorbored_dot_compat_dot_proto_dot_saver__pb2
+from tensorbored.compat.proto import struct_pb2 as tensorbored_dot_compat_dot_proto_dot_struct__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)tensorbored/compat/proto/meta_graph.proto\x12\x0btensorbored\x1a\x19google/protobuf/any.proto\x1a$tensorbored/compat/proto/graph.proto\x1a%tensorbored/compat/proto/op_def.proto\x1a%tensorbored/compat/proto/tensor.proto\x1a+tensorbored/compat/proto/tensor_shape.proto\x1a$tensorbored/compat/proto/types.proto\x1a\x31tensorbored/compat/proto/saved_object_graph.proto\x1a$tensorbored/compat/proto/saver.proto\x1a%tensorbored/compat/proto/struct.proto\"\xb3\x07\n\x0cMetaGraphDef\x12<\n\rmeta_info_def\x18\x01 \x01(\x0b\x32%.tensorbored.MetaGraphDef.MetaInfoDef\x12(\n\tgraph_def\x18\x02 \x01(\x0b\x32\x15.tensorbored.GraphDef\x12(\n\tsaver_def\x18\x03 \x01(\x0b\x32\x15.tensorbored.SaverDef\x12\x44\n\x0e\x63ollection_def\x18\x04 \x03(\x0b\x32,.tensorbored.MetaGraphDef.CollectionDefEntry\x12\x42\n\rsignature_def\x18\x05 \x03(\x0b\x32+.tensorbored.MetaGraphDef.SignatureDefEntry\x12\x31\n\x0e\x61sset_file_def\x18\x06 \x03(\x0b\x32\x19.tensorbored.AssetFileDef\x12\x37\n\x10object_graph_def\x18\x07 \x01(\x0b\x32\x1d.tensorbored.SavedObjectGraph\x1a\xf8\x02\n\x0bMetaInfoDef\x12\x1a\n\x12meta_graph_version\x18\x01 \x01(\t\x12-\n\x10stripped_op_list\x18\x02 \x01(\x0b\x32\x13.tensorbored.OpList\x12&\n\x08\x61ny_info\x18\x03 \x01(\x0b\x32\x14.google.protobuf.Any\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12\x1a\n\x12tensorflow_version\x18\x05 \x01(\t\x12\x1e\n\x16tensorflow_git_version\x18\x06 \x01(\t\x12\x1e\n\x16stripped_default_attrs\x18\x07 \x01(\x08\x12T\n\x10\x66unction_aliases\x18\x08 \x03(\x0b\x32:.tensorbored.MetaGraphDef.MetaInfoDef.FunctionAliasesEntry\x1a\x36\n\x14\x46unctionAliasesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1aP\n\x12\x43ollectionDefEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x1a.tensorbored.CollectionDef:\x02\x38\x01\x1aN\n\x11SignatureDefEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12(\n\x05value\x18\x02 \x01(\x0b\x32\x19.tensorbored.SignatureDef:\x02\x38\x01\"\xe4\x03\n\rCollectionDef\x12\x38\n\tnode_list\x18\x01 \x01(\x0b\x32#.tensorbored.CollectionDef.NodeListH\x00\x12:\n\nbytes_list\x18\x02 \x01(\x0b\x32$.tensorbored.CollectionDef.BytesListH\x00\x12:\n\nint64_list\x18\x03 \x01(\x0b\x32$.tensorbored.CollectionDef.Int64ListH\x00\x12:\n\nfloat_list\x18\x04 \x01(\x0b\x32$.tensorbored.CollectionDef.FloatListH\x00\x12\x36\n\x08\x61ny_list\x18\x05 \x01(\x0b\x32\".tensorbored.CollectionDef.AnyListH\x00\x1a\x19\n\x08NodeList\x12\r\n\x05value\x18\x01 \x03(\t\x1a\x1a\n\tBytesList\x12\r\n\x05value\x18\x01 \x03(\x0c\x1a\x1e\n\tInt64List\x12\x11\n\x05value\x18\x01 \x03(\x03\x42\x02\x10\x01\x1a\x1e\n\tFloatList\x12\x11\n\x05value\x18\x01 \x03(\x02\x42\x02\x10\x01\x1a.\n\x07\x41nyList\x12#\n\x05value\x18\x01 \x03(\x0b\x32\x14.google.protobuf.AnyB\x06\n\x04kind\"\xd7\x03\n\nTensorInfo\x12\x0e\n\x04name\x18\x01 \x01(\tH\x00\x12\x37\n\ncoo_sparse\x18\x04 \x01(\x0b\x32!.tensorbored.TensorInfo.CooSparseH\x00\x12\x43\n\x10\x63omposite_tensor\x18\x05 \x01(\x0b\x32\'.tensorbored.TensorInfo.CompositeTensorH\x00\x12$\n\x05\x64type\x18\x02 \x01(\x0e\x32\x15.tensorbored.DataType\x12\x33\n\x0ctensor_shape\x18\x03 \x01(\x0b\x32\x1d.tensorbored.TensorShapeProto\x1a\x65\n\tCooSparse\x12\x1a\n\x12values_tensor_name\x18\x01 \x01(\t\x12\x1b\n\x13indices_tensor_name\x18\x02 \x01(\t\x12\x1f\n\x17\x64\x65nse_shape_tensor_name\x18\x03 \x01(\t\x1am\n\x0f\x43ompositeTensor\x12-\n\ttype_spec\x18\x01 \x01(\x0b\x32\x1a.tensorbored.TypeSpecProto\x12+\n\ncomponents\x18\x02 \x03(\x0b\x32\x17.tensorbored.TensorInfoB\n\n\x08\x65ncoding\"\xaa\x03\n\x0cSignatureDef\x12\x35\n\x06inputs\x18\x01 \x03(\x0b\x32%.tensorbored.SignatureDef.InputsEntry\x12\x37\n\x07outputs\x18\x02 \x03(\x0b\x32&.tensorbored.SignatureDef.OutputsEntry\x12\x13\n\x0bmethod_name\x18\x03 \x01(\t\x12\x39\n\x08\x64\x65\x66\x61ults\x18\x04 \x03(\x0b\x32\'.tensorbored.SignatureDef.DefaultsEntry\x1a\x46\n\x0bInputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12&\n\x05value\x18\x02 \x01(\x0b\x32\x17.tensorbored.TensorInfo:\x02\x38\x01\x1aG\n\x0cOutputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12&\n\x05value\x18\x02 \x01(\x0b\x32\x17.tensorbored.TensorInfo:\x02\x38\x01\x1aI\n\rDefaultsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\'\n\x05value\x18\x02 \x01(\x0b\x32\x18.tensorbored.TensorProto:\x02\x38\x01\"N\n\x0c\x41ssetFileDef\x12,\n\x0btensor_info\x18\x01 \x01(\x0b\x32\x17.tensorbored.TensorInfo\x12\x10\n\x08\x66ilename\x18\x02 \x01(\tB\x87\x01\n\x18org.tensorflow.frameworkB\x0fMetaGraphProtosP\x01ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_METAGRAPHDEF = DESCRIPTOR.message_types_by_name['MetaGraphDef']
+_METAGRAPHDEF_METAINFODEF = _METAGRAPHDEF.nested_types_by_name['MetaInfoDef']
+_METAGRAPHDEF_METAINFODEF_FUNCTIONALIASESENTRY = _METAGRAPHDEF_METAINFODEF.nested_types_by_name['FunctionAliasesEntry']
+_METAGRAPHDEF_COLLECTIONDEFENTRY = _METAGRAPHDEF.nested_types_by_name['CollectionDefEntry']
+_METAGRAPHDEF_SIGNATUREDEFENTRY = _METAGRAPHDEF.nested_types_by_name['SignatureDefEntry']
+_COLLECTIONDEF = DESCRIPTOR.message_types_by_name['CollectionDef']
+_COLLECTIONDEF_NODELIST = _COLLECTIONDEF.nested_types_by_name['NodeList']
+_COLLECTIONDEF_BYTESLIST = _COLLECTIONDEF.nested_types_by_name['BytesList']
+_COLLECTIONDEF_INT64LIST = _COLLECTIONDEF.nested_types_by_name['Int64List']
+_COLLECTIONDEF_FLOATLIST = _COLLECTIONDEF.nested_types_by_name['FloatList']
+_COLLECTIONDEF_ANYLIST = _COLLECTIONDEF.nested_types_by_name['AnyList']
+_TENSORINFO = DESCRIPTOR.message_types_by_name['TensorInfo']
+_TENSORINFO_COOSPARSE = _TENSORINFO.nested_types_by_name['CooSparse']
+_TENSORINFO_COMPOSITETENSOR = _TENSORINFO.nested_types_by_name['CompositeTensor']
+_SIGNATUREDEF = DESCRIPTOR.message_types_by_name['SignatureDef']
+_SIGNATUREDEF_INPUTSENTRY = _SIGNATUREDEF.nested_types_by_name['InputsEntry']
+_SIGNATUREDEF_OUTPUTSENTRY = _SIGNATUREDEF.nested_types_by_name['OutputsEntry']
+_SIGNATUREDEF_DEFAULTSENTRY = _SIGNATUREDEF.nested_types_by_name['DefaultsEntry']
+_ASSETFILEDEF = DESCRIPTOR.message_types_by_name['AssetFileDef']
+MetaGraphDef = _reflection.GeneratedProtocolMessageType('MetaGraphDef', (_message.Message,), {
+
+  'MetaInfoDef' : _reflection.GeneratedProtocolMessageType('MetaInfoDef', (_message.Message,), {
+
+    'FunctionAliasesEntry' : _reflection.GeneratedProtocolMessageType('FunctionAliasesEntry', (_message.Message,), {
+      'DESCRIPTOR' : _METAGRAPHDEF_METAINFODEF_FUNCTIONALIASESENTRY,
+      '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.MetaGraphDef.MetaInfoDef.FunctionAliasesEntry)
+      })
+    ,
+    'DESCRIPTOR' : _METAGRAPHDEF_METAINFODEF,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.MetaGraphDef.MetaInfoDef)
+    })
+  ,
+
+  'CollectionDefEntry' : _reflection.GeneratedProtocolMessageType('CollectionDefEntry', (_message.Message,), {
+    'DESCRIPTOR' : _METAGRAPHDEF_COLLECTIONDEFENTRY,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.MetaGraphDef.CollectionDefEntry)
+    })
+  ,
+
+  'SignatureDefEntry' : _reflection.GeneratedProtocolMessageType('SignatureDefEntry', (_message.Message,), {
+    'DESCRIPTOR' : _METAGRAPHDEF_SIGNATUREDEFENTRY,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.MetaGraphDef.SignatureDefEntry)
+    })
+  ,
+  'DESCRIPTOR' : _METAGRAPHDEF,
+  '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.MetaGraphDef)
+  })
+_sym_db.RegisterMessage(MetaGraphDef)
+_sym_db.RegisterMessage(MetaGraphDef.MetaInfoDef)
+_sym_db.RegisterMessage(MetaGraphDef.MetaInfoDef.FunctionAliasesEntry)
+_sym_db.RegisterMessage(MetaGraphDef.CollectionDefEntry)
+_sym_db.RegisterMessage(MetaGraphDef.SignatureDefEntry)
+
+CollectionDef = _reflection.GeneratedProtocolMessageType('CollectionDef', (_message.Message,), {
+
+  'NodeList' : _reflection.GeneratedProtocolMessageType('NodeList', (_message.Message,), {
+    'DESCRIPTOR' : _COLLECTIONDEF_NODELIST,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CollectionDef.NodeList)
+    })
+  ,
+
+  'BytesList' : _reflection.GeneratedProtocolMessageType('BytesList', (_message.Message,), {
+    'DESCRIPTOR' : _COLLECTIONDEF_BYTESLIST,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CollectionDef.BytesList)
+    })
+  ,
+
+  'Int64List' : _reflection.GeneratedProtocolMessageType('Int64List', (_message.Message,), {
+    'DESCRIPTOR' : _COLLECTIONDEF_INT64LIST,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CollectionDef.Int64List)
+    })
+  ,
+
+  'FloatList' : _reflection.GeneratedProtocolMessageType('FloatList', (_message.Message,), {
+    'DESCRIPTOR' : _COLLECTIONDEF_FLOATLIST,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CollectionDef.FloatList)
+    })
+  ,
+
+  'AnyList' : _reflection.GeneratedProtocolMessageType('AnyList', (_message.Message,), {
+    'DESCRIPTOR' : _COLLECTIONDEF_ANYLIST,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CollectionDef.AnyList)
+    })
+  ,
+  'DESCRIPTOR' : _COLLECTIONDEF,
+  '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.CollectionDef)
+  })
+_sym_db.RegisterMessage(CollectionDef)
+_sym_db.RegisterMessage(CollectionDef.NodeList)
+_sym_db.RegisterMessage(CollectionDef.BytesList)
+_sym_db.RegisterMessage(CollectionDef.Int64List)
+_sym_db.RegisterMessage(CollectionDef.FloatList)
+_sym_db.RegisterMessage(CollectionDef.AnyList)
+
+TensorInfo = _reflection.GeneratedProtocolMessageType('TensorInfo', (_message.Message,), {
+
+  'CooSparse' : _reflection.GeneratedProtocolMessageType('CooSparse', (_message.Message,), {
+    'DESCRIPTOR' : _TENSORINFO_COOSPARSE,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.TensorInfo.CooSparse)
+    })
+  ,
+
+  'CompositeTensor' : _reflection.GeneratedProtocolMessageType('CompositeTensor', (_message.Message,), {
+    'DESCRIPTOR' : _TENSORINFO_COMPOSITETENSOR,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.TensorInfo.CompositeTensor)
+    })
+  ,
+  'DESCRIPTOR' : _TENSORINFO,
+  '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.TensorInfo)
+  })
+_sym_db.RegisterMessage(TensorInfo)
+_sym_db.RegisterMessage(TensorInfo.CooSparse)
+_sym_db.RegisterMessage(TensorInfo.CompositeTensor)
+
+SignatureDef = _reflection.GeneratedProtocolMessageType('SignatureDef', (_message.Message,), {
+
+  'InputsEntry' : _reflection.GeneratedProtocolMessageType('InputsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _SIGNATUREDEF_INPUTSENTRY,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.SignatureDef.InputsEntry)
+    })
+  ,
+
+  'OutputsEntry' : _reflection.GeneratedProtocolMessageType('OutputsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _SIGNATUREDEF_OUTPUTSENTRY,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.SignatureDef.OutputsEntry)
+    })
+  ,
+
+  'DefaultsEntry' : _reflection.GeneratedProtocolMessageType('DefaultsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _SIGNATUREDEF_DEFAULTSENTRY,
+    '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.SignatureDef.DefaultsEntry)
+    })
+  ,
+  'DESCRIPTOR' : _SIGNATUREDEF,
+  '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SignatureDef)
+  })
+_sym_db.RegisterMessage(SignatureDef)
+_sym_db.RegisterMessage(SignatureDef.InputsEntry)
+_sym_db.RegisterMessage(SignatureDef.OutputsEntry)
+_sym_db.RegisterMessage(SignatureDef.DefaultsEntry)
+
+AssetFileDef = _reflection.GeneratedProtocolMessageType('AssetFileDef', (_message.Message,), {
+  'DESCRIPTOR' : _ASSETFILEDEF,
+  '__module__' : 'tensorbored.compat.proto.meta_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.AssetFileDef)
+  })
+_sym_db.RegisterMessage(AssetFileDef)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\017MetaGraphProtosP\001ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\370\001\001'
+  _METAGRAPHDEF_METAINFODEF_FUNCTIONALIASESENTRY._options = None
+  _METAGRAPHDEF_METAINFODEF_FUNCTIONALIASESENTRY._serialized_options = b'8\001'
+  _METAGRAPHDEF_COLLECTIONDEFENTRY._options = None
+  _METAGRAPHDEF_COLLECTIONDEFENTRY._serialized_options = b'8\001'
+  _METAGRAPHDEF_SIGNATUREDEFENTRY._options = None
+  _METAGRAPHDEF_SIGNATUREDEFENTRY._serialized_options = b'8\001'
+  _COLLECTIONDEF_INT64LIST.fields_by_name['value']._options = None
+  _COLLECTIONDEF_INT64LIST.fields_by_name['value']._serialized_options = b'\020\001'
+  _COLLECTIONDEF_FLOATLIST.fields_by_name['value']._options = None
+  _COLLECTIONDEF_FLOATLIST.fields_by_name['value']._serialized_options = b'\020\001'
+  _SIGNATUREDEF_INPUTSENTRY._options = None
+  _SIGNATUREDEF_INPUTSENTRY._serialized_options = b'8\001'
+  _SIGNATUREDEF_OUTPUTSENTRY._options = None
+  _SIGNATUREDEF_OUTPUTSENTRY._serialized_options = b'8\001'
+  _SIGNATUREDEF_DEFAULTSENTRY._options = None
+  _SIGNATUREDEF_DEFAULTSENTRY._serialized_options = b'8\001'
+  _METAGRAPHDEF._serialized_start=413
+  _METAGRAPHDEF._serialized_end=1360
+  _METAGRAPHDEF_METAINFODEF._serialized_start=822
+  _METAGRAPHDEF_METAINFODEF._serialized_end=1198
+  _METAGRAPHDEF_METAINFODEF_FUNCTIONALIASESENTRY._serialized_start=1144
+  _METAGRAPHDEF_METAINFODEF_FUNCTIONALIASESENTRY._serialized_end=1198
+  _METAGRAPHDEF_COLLECTIONDEFENTRY._serialized_start=1200
+  _METAGRAPHDEF_COLLECTIONDEFENTRY._serialized_end=1280
+  _METAGRAPHDEF_SIGNATUREDEFENTRY._serialized_start=1282
+  _METAGRAPHDEF_SIGNATUREDEFENTRY._serialized_end=1360
+  _COLLECTIONDEF._serialized_start=1363
+  _COLLECTIONDEF._serialized_end=1847
+  _COLLECTIONDEF_NODELIST._serialized_start=1674
+  _COLLECTIONDEF_NODELIST._serialized_end=1699
+  _COLLECTIONDEF_BYTESLIST._serialized_start=1701
+  _COLLECTIONDEF_BYTESLIST._serialized_end=1727
+  _COLLECTIONDEF_INT64LIST._serialized_start=1729
+  _COLLECTIONDEF_INT64LIST._serialized_end=1759
+  _COLLECTIONDEF_FLOATLIST._serialized_start=1761
+  _COLLECTIONDEF_FLOATLIST._serialized_end=1791
+  _COLLECTIONDEF_ANYLIST._serialized_start=1793
+  _COLLECTIONDEF_ANYLIST._serialized_end=1839
+  _TENSORINFO._serialized_start=1850
+  _TENSORINFO._serialized_end=2321
+  _TENSORINFO_COOSPARSE._serialized_start=2097
+  _TENSORINFO_COOSPARSE._serialized_end=2198
+  _TENSORINFO_COMPOSITETENSOR._serialized_start=2200
+  _TENSORINFO_COMPOSITETENSOR._serialized_end=2309
+  _SIGNATUREDEF._serialized_start=2324
+  _SIGNATUREDEF._serialized_end=2750
+  _SIGNATUREDEF_INPUTSENTRY._serialized_start=2532
+  _SIGNATUREDEF_INPUTSENTRY._serialized_end=2602
+  _SIGNATUREDEF_OUTPUTSENTRY._serialized_start=2604
+  _SIGNATUREDEF_OUTPUTSENTRY._serialized_end=2675
+  _SIGNATUREDEF_DEFAULTSENTRY._serialized_start=2677
+  _SIGNATUREDEF_DEFAULTSENTRY._serialized_end=2750
+  _ASSETFILEDEF._serialized_start=2752
+  _ASSETFILEDEF._serialized_end=2830
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/node_def_pb2.py

@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/node_def.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 full_type_pb2 as tensorbored_dot_compat_dot_proto_dot_full__type__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'tensorbored/compat/proto/node_def.proto\x12\x0btensorbored\x1a)tensorbored/compat/proto/attr_value.proto\x1a(tensorbored/compat/proto/full_type.proto\"\x8a\x03\n\x07NodeDef\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\n\n\x02op\x18\x02 \x01(\t\x12\r\n\x05input\x18\x03 \x03(\t\x12\x0e\n\x06\x64\x65vice\x18\x04 \x01(\t\x12,\n\x04\x61ttr\x18\x05 \x03(\x0b\x32\x1e.tensorbored.NodeDef.AttrEntry\x12K\n\x17\x65xperimental_debug_info\x18\x06 \x01(\x0b\x32*.tensorbored.NodeDef.ExperimentalDebugInfo\x12\x33\n\x11\x65xperimental_type\x18\x07 \x01(\x0b\x32\x18.tensorbored.FullTypeDef\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\x15\x45xperimentalDebugInfo\x12\x1b\n\x13original_node_names\x18\x01 \x03(\t\x12\x1b\n\x13original_func_names\x18\x02 \x03(\tB{\n\x18org.tensorflow.frameworkB\tNodeProtoP\x01ZOgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/node_def_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_NODEDEF = DESCRIPTOR.message_types_by_name['NodeDef']
+_NODEDEF_ATTRENTRY = _NODEDEF.nested_types_by_name['AttrEntry']
+_NODEDEF_EXPERIMENTALDEBUGINFO = _NODEDEF.nested_types_by_name['ExperimentalDebugInfo']
+NodeDef = _reflection.GeneratedProtocolMessageType('NodeDef', (_message.Message,), {
+
+  'AttrEntry' : _reflection.GeneratedProtocolMessageType('AttrEntry', (_message.Message,), {
+    'DESCRIPTOR' : _NODEDEF_ATTRENTRY,
+    '__module__' : 'tensorbored.compat.proto.node_def_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.NodeDef.AttrEntry)
+    })
+  ,
+
+  'ExperimentalDebugInfo' : _reflection.GeneratedProtocolMessageType('ExperimentalDebugInfo', (_message.Message,), {
+    'DESCRIPTOR' : _NODEDEF_EXPERIMENTALDEBUGINFO,
+    '__module__' : 'tensorbored.compat.proto.node_def_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.NodeDef.ExperimentalDebugInfo)
+    })
+  ,
+  'DESCRIPTOR' : _NODEDEF,
+  '__module__' : 'tensorbored.compat.proto.node_def_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.NodeDef)
+  })
+_sym_db.RegisterMessage(NodeDef)
+_sym_db.RegisterMessage(NodeDef.AttrEntry)
+_sym_db.RegisterMessage(NodeDef.ExperimentalDebugInfo)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\tNodeProtoP\001ZOgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/node_def_go_proto\370\001\001'
+  _NODEDEF_ATTRENTRY._options = None
+  _NODEDEF_ATTRENTRY._serialized_options = b'8\001'
+  _NODEDEF._serialized_start=142
+  _NODEDEF._serialized_end=536
+  _NODEDEF_ATTRENTRY._serialized_start=386
+  _NODEDEF_ATTRENTRY._serialized_end=453
+  _NODEDEF_EXPERIMENTALDEBUGINFO._serialized_start=455
+  _NODEDEF_EXPERIMENTALDEBUGINFO._serialized_end=536
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/op_def_pb2.py

@@ -0,0 +1,81 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/op_def.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 full_type_pb2 as tensorbored_dot_compat_dot_proto_dot_full__type__pb2
+from tensorbored.compat.proto import resource_handle_pb2 as tensorbored_dot_compat_dot_proto_dot_resource__handle__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/op_def.proto\x12\x0btensorbored\x1a)tensorbored/compat/proto/attr_value.proto\x1a(tensorbored/compat/proto/full_type.proto\x1a.tensorbored/compat/proto/resource_handle.proto\x1a$tensorbored/compat/proto/types.proto\"\xfc\x06\n\x05OpDef\x12\x0c\n\x04name\x18\x01 \x01(\t\x12,\n\tinput_arg\x18\x02 \x03(\x0b\x32\x19.tensorbored.OpDef.ArgDef\x12-\n\noutput_arg\x18\x03 \x03(\x0b\x32\x19.tensorbored.OpDef.ArgDef\x12\x16\n\x0e\x63ontrol_output\x18\x14 \x03(\t\x12(\n\x04\x61ttr\x18\x04 \x03(\x0b\x32\x1a.tensorbored.OpDef.AttrDef\x12/\n\x0b\x64\x65precation\x18\x08 \x01(\x0b\x32\x1a.tensorbored.OpDeprecation\x12\x0f\n\x07summary\x18\x05 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x06 \x01(\t\x12\x16\n\x0eis_commutative\x18\x12 \x01(\x08\x12\x14\n\x0cis_aggregate\x18\x10 \x01(\x08\x12\x13\n\x0bis_stateful\x18\x11 \x01(\x08\x12\"\n\x1a\x61llows_uninitialized_input\x18\x13 \x01(\x08\x12$\n\x1cis_distributed_communication\x18\x15 \x01(\x08\x1a\x9f\x02\n\x06\x41rgDef\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12#\n\x04type\x18\x03 \x01(\x0e\x32\x15.tensorbored.DataType\x12\x11\n\ttype_attr\x18\x04 \x01(\t\x12\x13\n\x0bnumber_attr\x18\x05 \x01(\t\x12\x16\n\x0etype_list_attr\x18\x06 \x01(\t\x12\x43\n\x0bhandle_data\x18\x07 \x03(\x0b\x32..tensorbored.ResourceHandleProto.DtypeAndShape\x12\x0e\n\x06is_ref\x18\x10 \x01(\x08\x12\x38\n\x16\x65xperimental_full_type\x18\x11 \x01(\x0b\x32\x18.tensorbored.FullTypeDef\x1a\xbf\x01\n\x07\x41ttrDef\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04type\x18\x02 \x01(\t\x12-\n\rdefault_value\x18\x03 \x01(\x0b\x32\x16.tensorbored.AttrValue\x12\x13\n\x0b\x64\x65scription\x18\x04 \x01(\t\x12\x13\n\x0bhas_minimum\x18\x05 \x01(\x08\x12\x0f\n\x07minimum\x18\x06 \x01(\x03\x12.\n\x0e\x61llowed_values\x18\x07 \x01(\x0b\x32\x16.tensorbored.AttrValue\"5\n\rOpDeprecation\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12\x13\n\x0b\x65xplanation\x18\x02 \x01(\t\"(\n\x06OpList\x12\x1e\n\x02op\x18\x01 \x03(\x0b\x32\x12.tensorbored.OpDefB{\n\x18org.tensorflow.frameworkB\x0bOpDefProtosP\x01ZMgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/op_def_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_OPDEF = DESCRIPTOR.message_types_by_name['OpDef']
+_OPDEF_ARGDEF = _OPDEF.nested_types_by_name['ArgDef']
+_OPDEF_ATTRDEF = _OPDEF.nested_types_by_name['AttrDef']
+_OPDEPRECATION = DESCRIPTOR.message_types_by_name['OpDeprecation']
+_OPLIST = DESCRIPTOR.message_types_by_name['OpList']
+OpDef = _reflection.GeneratedProtocolMessageType('OpDef', (_message.Message,), {
+
+  'ArgDef' : _reflection.GeneratedProtocolMessageType('ArgDef', (_message.Message,), {
+    'DESCRIPTOR' : _OPDEF_ARGDEF,
+    '__module__' : 'tensorbored.compat.proto.op_def_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.OpDef.ArgDef)
+    })
+  ,
+
+  'AttrDef' : _reflection.GeneratedProtocolMessageType('AttrDef', (_message.Message,), {
+    'DESCRIPTOR' : _OPDEF_ATTRDEF,
+    '__module__' : 'tensorbored.compat.proto.op_def_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.OpDef.AttrDef)
+    })
+  ,
+  'DESCRIPTOR' : _OPDEF,
+  '__module__' : 'tensorbored.compat.proto.op_def_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.OpDef)
+  })
+_sym_db.RegisterMessage(OpDef)
+_sym_db.RegisterMessage(OpDef.ArgDef)
+_sym_db.RegisterMessage(OpDef.AttrDef)
+
+OpDeprecation = _reflection.GeneratedProtocolMessageType('OpDeprecation', (_message.Message,), {
+  'DESCRIPTOR' : _OPDEPRECATION,
+  '__module__' : 'tensorbored.compat.proto.op_def_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.OpDeprecation)
+  })
+_sym_db.RegisterMessage(OpDeprecation)
+
+OpList = _reflection.GeneratedProtocolMessageType('OpList', (_message.Message,), {
+  'DESCRIPTOR' : _OPLIST,
+  '__module__' : 'tensorbored.compat.proto.op_def_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.OpList)
+  })
+_sym_db.RegisterMessage(OpList)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\013OpDefProtosP\001ZMgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/op_def_go_proto\370\001\001'
+  _OPDEF._serialized_start=226
+  _OPDEF._serialized_end=1118
+  _OPDEF_ARGDEF._serialized_start=637
+  _OPDEF_ARGDEF._serialized_end=924
+  _OPDEF_ATTRDEF._serialized_start=927
+  _OPDEF_ATTRDEF._serialized_end=1118
+  _OPDEPRECATION._serialized_start=1120
+  _OPDEPRECATION._serialized_end=1173
+  _OPLIST._serialized_start=1175
+  _OPLIST._serialized_end=1215
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/resource_handle_pb2.py

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/resource_handle.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_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/resource_handle.proto\x12\x0btensorbored\x1a+tensorbored/compat/proto/tensor_shape.proto\x1a$tensorbored/compat/proto/types.proto\"\xa8\x02\n\x13ResourceHandleProto\x12\x0e\n\x06\x64\x65vice\x18\x01 \x01(\t\x12\x11\n\tcontainer\x18\x02 \x01(\t\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x11\n\thash_code\x18\x04 \x01(\x04\x12\x17\n\x0fmaybe_type_name\x18\x05 \x01(\t\x12I\n\x11\x64types_and_shapes\x18\x06 \x03(\x0b\x32..tensorbored.ResourceHandleProto.DtypeAndShape\x1a\x63\n\rDtypeAndShape\x12$\n\x05\x64type\x18\x01 \x01(\x0e\x32\x15.tensorbored.DataType\x12,\n\x05shape\x18\x02 \x01(\x0b\x32\x1d.tensorbored.TensorShapeProtoJ\x04\x08\x07\x10\x08\x42\x87\x01\n\x18org.tensorflow.frameworkB\x0eResourceHandleP\x01ZVgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/resource_handle_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_RESOURCEHANDLEPROTO = DESCRIPTOR.message_types_by_name['ResourceHandleProto']
+_RESOURCEHANDLEPROTO_DTYPEANDSHAPE = _RESOURCEHANDLEPROTO.nested_types_by_name['DtypeAndShape']
+ResourceHandleProto = _reflection.GeneratedProtocolMessageType('ResourceHandleProto', (_message.Message,), {
+
+  'DtypeAndShape' : _reflection.GeneratedProtocolMessageType('DtypeAndShape', (_message.Message,), {
+    'DESCRIPTOR' : _RESOURCEHANDLEPROTO_DTYPEANDSHAPE,
+    '__module__' : 'tensorbored.compat.proto.resource_handle_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ResourceHandleProto.DtypeAndShape)
+    })
+  ,
+  'DESCRIPTOR' : _RESOURCEHANDLEPROTO,
+  '__module__' : 'tensorbored.compat.proto.resource_handle_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ResourceHandleProto)
+  })
+_sym_db.RegisterMessage(ResourceHandleProto)
+_sym_db.RegisterMessage(ResourceHandleProto.DtypeAndShape)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\016ResourceHandleP\001ZVgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/resource_handle_go_proto\370\001\001'
+  _RESOURCEHANDLEPROTO._serialized_start=147
+  _RESOURCEHANDLEPROTO._serialized_end=443
+  _RESOURCEHANDLEPROTO_DTYPEANDSHAPE._serialized_start=338
+  _RESOURCEHANDLEPROTO_DTYPEANDSHAPE._serialized_end=437
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/rewriter_config_pb2.py

@@ -0,0 +1,93 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/rewriter_config.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 verifier_config_pb2 as tensorbored_dot_compat_dot_proto_dot_verifier__config__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n.tensorbored/compat/proto/rewriter_config.proto\x12\x0btensorbored\x1a)tensorbored/compat/proto/attr_value.proto\x1a.tensorbored/compat/proto/verifier_config.proto\";\n\x13\x41utoParallelOptions\x12\x0e\n\x06\x65nable\x18\x01 \x01(\x08\x12\x14\n\x0cnum_replicas\x18\x02 \x01(\x05\"+\n\x16ScopedAllocatorOptions\x12\x11\n\tenable_op\x18\x01 \x03(\t\"\xb2\x16\n\x0eRewriterConfig\x12\x44\n\x15\x63pu_layout_conversion\x18\x32 \x01(\x0e\x32%.tensorbored.RewriterConfig.CpuLayout\x12<\n\x10layout_optimizer\x18\x01 \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12<\n\x10\x63onstant_folding\x18\x03 \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12>\n\x12shape_optimization\x18\r \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12\x35\n\tremapping\x18\x0e \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12G\n\x1b\x63ommon_subgraph_elimination\x18\x18 \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12\x43\n\x17\x61rithmetic_optimization\x18\x07 \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12\x43\n\x17\x64\x65pendency_optimization\x18\x08 \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12=\n\x11loop_optimization\x18\t \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12\x41\n\x15\x66unction_optimization\x18\n \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12:\n\x0e\x64\x65\x62ug_stripper\x18\x0b \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12\x1d\n\x15\x64isable_model_pruning\x18\x02 \x01(\x08\x12I\n\x1dscoped_allocator_optimization\x18\x0f \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12\x44\n\x18pin_to_host_optimization\x18\x12 \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12\x43\n\x17implementation_selector\x18\x16 \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12@\n\x14\x61uto_mixed_precision\x18\x17 \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12\x44\n\x18\x61uto_mixed_precision_mkl\x18\x19 \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12P\n$auto_mixed_precision_onednn_bfloat16\x18\x1f \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12\x44\n\x18\x61uto_mixed_precision_cpu\x18\x1d \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12\x1e\n\x16\x64isable_meta_optimizer\x18\x13 \x01(\x08\x12\x1d\n\x15\x64isable_tfg_optimizer\x18  \x01(\x08\x12\x41\n\x15use_plugin_optimizers\x18\x1c \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12P\n$experimental_conditional_code_motion\x18\x1e \x01(\x0e\x32\".tensorbored.RewriterConfig.Toggle\x12P\n\x19meta_optimizer_iterations\x18\x0c \x01(\x0e\x32-.tensorbored.RewriterConfig.NumIterationsType\x12\x17\n\x0fmin_graph_nodes\x18\x11 \x01(\x05\x12;\n3experimental_disable_compressed_tensor_optimization\x18\x1a \x01(\x08\x12;\n3experimental_disable_folding_quantization_emulation\x18\x1b \x01(\x08\x12\x43\n\x13memory_optimization\x18\x04 \x01(\x0e\x32&.tensorbored.RewriterConfig.MemOptType\x12/\n\'memory_optimizer_target_node_name_scope\x18\x06 \x01(\t\x12!\n\x19meta_optimizer_timeout_ms\x18\x14 \x01(\x03\x12\x37\n\rauto_parallel\x18\x05 \x01(\x0b\x32 .tensorbored.AutoParallelOptions\x12 \n\x18\x66\x61il_on_optimizer_errors\x18\x15 \x01(\x08\x12\x42\n\x15scoped_allocator_opts\x18\x10 \x01(\x0b\x32#.tensorbored.ScopedAllocatorOptions\x12\x12\n\noptimizers\x18\x64 \x03(\t\x12L\n\x11\x63ustom_optimizers\x18\xc8\x01 \x03(\x0b\x32\x30.tensorbored.RewriterConfig.CustomGraphOptimizer\x12\x45\n\x1finter_optimizer_verifier_config\x18\xac\x02 \x01(\x0b\x32\x1b.tensorbored.VerifierConfig\x12G\n!post_optimization_verifier_config\x18\xad\x02 \x01(\x0b\x32\x1b.tensorbored.VerifierConfig\x1a\xcc\x01\n\x14\x43ustomGraphOptimizer\x12\x0c\n\x04name\x18\x01 \x01(\t\x12Y\n\rparameter_map\x18\x02 \x03(\x0b\x32\x42.tensorbored.RewriterConfig.CustomGraphOptimizer.ParameterMapEntry\x1aK\n\x11ParameterMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.tensorbored.AttrValue:\x02\x38\x01\"d\n\x06Toggle\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x06\n\x02ON\x10\x01\x12\x07\n\x03OFF\x10\x02\x12\x0e\n\nAGGRESSIVE\x10\x03\x12\x15\n\x11\x45XPERIMENTAL_MLIR\x10\x04\x12\x15\n\x11\x45XPERIMENTAL_BOTH\x10\x05\"I\n\tCpuLayout\x12\x18\n\x14NO_CONVERSION_ON_CPU\x10\x00\x12\x10\n\x0cNCHW_TO_NHWC\x10\x01\x12\x10\n\x0cNHWC_TO_NCHW\x10\x02\"<\n\x11NumIterationsType\x12\x15\n\x11\x44\x45\x46\x41ULT_NUM_ITERS\x10\x00\x12\x07\n\x03ONE\x10\x01\x12\x07\n\x03TWO\x10\x02\"\x9f\x01\n\nMemOptType\x12\x13\n\x0f\x44\x45\x46\x41ULT_MEM_OPT\x10\x00\x12\x0e\n\nNO_MEM_OPT\x10\x01\x12\n\n\x06MANUAL\x10\x02\x12\x17\n\x13SWAPPING_HEURISTICS\x10\x04\x12\x1c\n\x18RECOMPUTATION_HEURISTICS\x10\x05\x12\x19\n\x15SCHEDULING_HEURISTICS\x10\x06\x12\x0e\n\nHEURISTICS\x10\x03\x42\x8c\x01\n\x18org.tensorflow.frameworkB\x14RewriterConfigProtosP\x01ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_AUTOPARALLELOPTIONS = DESCRIPTOR.message_types_by_name['AutoParallelOptions']
+_SCOPEDALLOCATOROPTIONS = DESCRIPTOR.message_types_by_name['ScopedAllocatorOptions']
+_REWRITERCONFIG = DESCRIPTOR.message_types_by_name['RewriterConfig']
+_REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER = _REWRITERCONFIG.nested_types_by_name['CustomGraphOptimizer']
+_REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER_PARAMETERMAPENTRY = _REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER.nested_types_by_name['ParameterMapEntry']
+_REWRITERCONFIG_TOGGLE = _REWRITERCONFIG.enum_types_by_name['Toggle']
+_REWRITERCONFIG_CPULAYOUT = _REWRITERCONFIG.enum_types_by_name['CpuLayout']
+_REWRITERCONFIG_NUMITERATIONSTYPE = _REWRITERCONFIG.enum_types_by_name['NumIterationsType']
+_REWRITERCONFIG_MEMOPTTYPE = _REWRITERCONFIG.enum_types_by_name['MemOptType']
+AutoParallelOptions = _reflection.GeneratedProtocolMessageType('AutoParallelOptions', (_message.Message,), {
+  'DESCRIPTOR' : _AUTOPARALLELOPTIONS,
+  '__module__' : 'tensorbored.compat.proto.rewriter_config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.AutoParallelOptions)
+  })
+_sym_db.RegisterMessage(AutoParallelOptions)
+
+ScopedAllocatorOptions = _reflection.GeneratedProtocolMessageType('ScopedAllocatorOptions', (_message.Message,), {
+  'DESCRIPTOR' : _SCOPEDALLOCATOROPTIONS,
+  '__module__' : 'tensorbored.compat.proto.rewriter_config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ScopedAllocatorOptions)
+  })
+_sym_db.RegisterMessage(ScopedAllocatorOptions)
+
+RewriterConfig = _reflection.GeneratedProtocolMessageType('RewriterConfig', (_message.Message,), {
+
+  'CustomGraphOptimizer' : _reflection.GeneratedProtocolMessageType('CustomGraphOptimizer', (_message.Message,), {
+
+    'ParameterMapEntry' : _reflection.GeneratedProtocolMessageType('ParameterMapEntry', (_message.Message,), {
+      'DESCRIPTOR' : _REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER_PARAMETERMAPENTRY,
+      '__module__' : 'tensorbored.compat.proto.rewriter_config_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.RewriterConfig.CustomGraphOptimizer.ParameterMapEntry)
+      })
+    ,
+    'DESCRIPTOR' : _REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER,
+    '__module__' : 'tensorbored.compat.proto.rewriter_config_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.RewriterConfig.CustomGraphOptimizer)
+    })
+  ,
+  'DESCRIPTOR' : _REWRITERCONFIG,
+  '__module__' : 'tensorbored.compat.proto.rewriter_config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.RewriterConfig)
+  })
+_sym_db.RegisterMessage(RewriterConfig)
+_sym_db.RegisterMessage(RewriterConfig.CustomGraphOptimizer)
+_sym_db.RegisterMessage(RewriterConfig.CustomGraphOptimizer.ParameterMapEntry)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\024RewriterConfigProtosP\001ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\370\001\001'
+  _REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER_PARAMETERMAPENTRY._options = None
+  _REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER_PARAMETERMAPENTRY._serialized_options = b'8\001'
+  _AUTOPARALLELOPTIONS._serialized_start=154
+  _AUTOPARALLELOPTIONS._serialized_end=213
+  _SCOPEDALLOCATOROPTIONS._serialized_start=215
+  _SCOPEDALLOCATOROPTIONS._serialized_end=258
+  _REWRITERCONFIG._serialized_start=261
+  _REWRITERCONFIG._serialized_end=3127
+  _REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER._serialized_start=2522
+  _REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER._serialized_end=2726
+  _REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER_PARAMETERMAPENTRY._serialized_start=2651
+  _REWRITERCONFIG_CUSTOMGRAPHOPTIMIZER_PARAMETERMAPENTRY._serialized_end=2726
+  _REWRITERCONFIG_TOGGLE._serialized_start=2728
+  _REWRITERCONFIG_TOGGLE._serialized_end=2828
+  _REWRITERCONFIG_CPULAYOUT._serialized_start=2830
+  _REWRITERCONFIG_CPULAYOUT._serialized_end=2903
+  _REWRITERCONFIG_NUMITERATIONSTYPE._serialized_start=2905
+  _REWRITERCONFIG_NUMITERATIONSTYPE._serialized_end=2965
+  _REWRITERCONFIG_MEMOPTTYPE._serialized_start=2968
+  _REWRITERCONFIG_MEMOPTTYPE._serialized_end=3127
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/rpc_options_pb2.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/rpc_options.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/rpc_options.proto\x12\x0btensorbored\"\xd5\x01\n\nRPCOptions\x12$\n\x1cuse_rpc_for_inprocess_master\x18\x01 \x01(\x08\x12\x1d\n\x15\x63ompression_algorithm\x18\x02 \x01(\t\x12\x19\n\x11\x63ompression_level\x18\x03 \x01(\x05\x12\x1a\n\x12\x63\x61\x63he_rpc_response\x18\x04 \x01(\x08\x12*\n\"disable_session_connection_sharing\x18\x05 \x01(\x08\x12\x1f\n\x17num_channels_per_target\x18\x06 \x01(\x05\x42@Z>github.com/google/tsl/tsl/go/protobuf/for_core_protos_go_protob\x06proto3')
+
+
+
+_RPCOPTIONS = DESCRIPTOR.message_types_by_name['RPCOptions']
+RPCOptions = _reflection.GeneratedProtocolMessageType('RPCOptions', (_message.Message,), {
+  'DESCRIPTOR' : _RPCOPTIONS,
+  '__module__' : 'tensorbored.compat.proto.rpc_options_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.RPCOptions)
+  })
+_sym_db.RegisterMessage(RPCOptions)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'Z>github.com/google/tsl/tsl/go/protobuf/for_core_protos_go_proto'
+  _RPCOPTIONS._serialized_start=60
+  _RPCOPTIONS._serialized_end=273
+# @@protoc_insertion_point(module_scope)