tensorbored 2.21.0rc1769983804__py3-none-any.whl

tensorbored/compat/proto/attr_value_pb2.py

@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/attr_value.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/attr_value.proto\x12\x0btensorbored\x1a%tensorbored/compat/proto/tensor.proto\x1a+tensorbored/compat/proto/tensor_shape.proto\x1a$tensorbored/compat/proto/types.proto\"\xaf\x04\n\tAttrValue\x12\x0b\n\x01s\x18\x02 \x01(\x0cH\x00\x12\x0b\n\x01i\x18\x03 \x01(\x03H\x00\x12\x0b\n\x01\x66\x18\x04 \x01(\x02H\x00\x12\x0b\n\x01\x62\x18\x05 \x01(\x08H\x00\x12%\n\x04type\x18\x06 \x01(\x0e\x32\x15.tensorbored.DataTypeH\x00\x12.\n\x05shape\x18\x07 \x01(\x0b\x32\x1d.tensorbored.TensorShapeProtoH\x00\x12*\n\x06tensor\x18\x08 \x01(\x0b\x32\x18.tensorbored.TensorProtoH\x00\x12\x30\n\x04list\x18\x01 \x01(\x0b\x32 .tensorbored.AttrValue.ListValueH\x00\x12)\n\x04\x66unc\x18\n \x01(\x0b\x32\x19.tensorbored.NameAttrListH\x00\x12\x15\n\x0bplaceholder\x18\t \x01(\tH\x00\x1a\xed\x01\n\tListValue\x12\t\n\x01s\x18\x02 \x03(\x0c\x12\r\n\x01i\x18\x03 \x03(\x03\x42\x02\x10\x01\x12\r\n\x01\x66\x18\x04 \x03(\x02\x42\x02\x10\x01\x12\r\n\x01\x62\x18\x05 \x03(\x08\x42\x02\x10\x01\x12\'\n\x04type\x18\x06 \x03(\x0e\x32\x15.tensorbored.DataTypeB\x02\x10\x01\x12,\n\x05shape\x18\x07 \x03(\x0b\x32\x1d.tensorbored.TensorShapeProto\x12(\n\x06tensor\x18\x08 \x03(\x0b\x32\x18.tensorbored.TensorProto\x12\'\n\x04\x66unc\x18\t \x03(\x0b\x32\x19.tensorbored.NameAttrListB\x07\n\x05value\"\x94\x01\n\x0cNameAttrList\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x31\n\x04\x61ttr\x18\x02 \x03(\x0b\x32#.tensorbored.NameAttrList.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\x42\x83\x01\n\x18org.tensorflow.frameworkB\x0f\x41ttrValueProtosP\x01ZQgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/attr_value_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_ATTRVALUE = DESCRIPTOR.message_types_by_name['AttrValue']
+_ATTRVALUE_LISTVALUE = _ATTRVALUE.nested_types_by_name['ListValue']
+_NAMEATTRLIST = DESCRIPTOR.message_types_by_name['NameAttrList']
+_NAMEATTRLIST_ATTRENTRY = _NAMEATTRLIST.nested_types_by_name['AttrEntry']
+AttrValue = _reflection.GeneratedProtocolMessageType('AttrValue', (_message.Message,), {
+
+  'ListValue' : _reflection.GeneratedProtocolMessageType('ListValue', (_message.Message,), {
+    'DESCRIPTOR' : _ATTRVALUE_LISTVALUE,
+    '__module__' : 'tensorbored.compat.proto.attr_value_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.AttrValue.ListValue)
+    })
+  ,
+  'DESCRIPTOR' : _ATTRVALUE,
+  '__module__' : 'tensorbored.compat.proto.attr_value_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.AttrValue)
+  })
+_sym_db.RegisterMessage(AttrValue)
+_sym_db.RegisterMessage(AttrValue.ListValue)
+
+NameAttrList = _reflection.GeneratedProtocolMessageType('NameAttrList', (_message.Message,), {
+
+  'AttrEntry' : _reflection.GeneratedProtocolMessageType('AttrEntry', (_message.Message,), {
+    'DESCRIPTOR' : _NAMEATTRLIST_ATTRENTRY,
+    '__module__' : 'tensorbored.compat.proto.attr_value_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.NameAttrList.AttrEntry)
+    })
+  ,
+  'DESCRIPTOR' : _NAMEATTRLIST,
+  '__module__' : 'tensorbored.compat.proto.attr_value_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.NameAttrList)
+  })
+_sym_db.RegisterMessage(NameAttrList)
+_sym_db.RegisterMessage(NameAttrList.AttrEntry)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\017AttrValueProtosP\001ZQgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/attr_value_go_proto\370\001\001'
+  _ATTRVALUE_LISTVALUE.fields_by_name['i']._options = None
+  _ATTRVALUE_LISTVALUE.fields_by_name['i']._serialized_options = b'\020\001'
+  _ATTRVALUE_LISTVALUE.fields_by_name['f']._options = None
+  _ATTRVALUE_LISTVALUE.fields_by_name['f']._serialized_options = b'\020\001'
+  _ATTRVALUE_LISTVALUE.fields_by_name['b']._options = None
+  _ATTRVALUE_LISTVALUE.fields_by_name['b']._serialized_options = b'\020\001'
+  _ATTRVALUE_LISTVALUE.fields_by_name['type']._options = None
+  _ATTRVALUE_LISTVALUE.fields_by_name['type']._serialized_options = b'\020\001'
+  _NAMEATTRLIST_ATTRENTRY._options = None
+  _NAMEATTRLIST_ATTRENTRY._serialized_options = b'8\001'
+  _ATTRVALUE._serialized_start=181
+  _ATTRVALUE._serialized_end=740
+  _ATTRVALUE_LISTVALUE._serialized_start=494
+  _ATTRVALUE_LISTVALUE._serialized_end=731
+  _NAMEATTRLIST._serialized_start=743
+  _NAMEATTRLIST._serialized_end=891
+  _NAMEATTRLIST_ATTRENTRY._serialized_start=824
+  _NAMEATTRLIST_ATTRENTRY._serialized_end=891
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/cluster_pb2.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/cluster.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/cluster.proto\x12\x0btensorbored\"s\n\x06JobDef\x12\x0c\n\x04name\x18\x01 \x01(\t\x12-\n\x05tasks\x18\x02 \x03(\x0b\x32\x1e.tensorbored.JobDef.TasksEntry\x1a,\n\nTasksEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\".\n\nClusterDef\x12 \n\x03job\x18\x01 \x03(\x0b\x32\x13.tensorbored.JobDefB\x87\x01\n\x1aorg.tensorflow.distruntimeB\rClusterProtosP\x01ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_JOBDEF = DESCRIPTOR.message_types_by_name['JobDef']
+_JOBDEF_TASKSENTRY = _JOBDEF.nested_types_by_name['TasksEntry']
+_CLUSTERDEF = DESCRIPTOR.message_types_by_name['ClusterDef']
+JobDef = _reflection.GeneratedProtocolMessageType('JobDef', (_message.Message,), {
+
+  'TasksEntry' : _reflection.GeneratedProtocolMessageType('TasksEntry', (_message.Message,), {
+    'DESCRIPTOR' : _JOBDEF_TASKSENTRY,
+    '__module__' : 'tensorbored.compat.proto.cluster_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.JobDef.TasksEntry)
+    })
+  ,
+  'DESCRIPTOR' : _JOBDEF,
+  '__module__' : 'tensorbored.compat.proto.cluster_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.JobDef)
+  })
+_sym_db.RegisterMessage(JobDef)
+_sym_db.RegisterMessage(JobDef.TasksEntry)
+
+ClusterDef = _reflection.GeneratedProtocolMessageType('ClusterDef', (_message.Message,), {
+  'DESCRIPTOR' : _CLUSTERDEF,
+  '__module__' : 'tensorbored.compat.proto.cluster_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ClusterDef)
+  })
+_sym_db.RegisterMessage(ClusterDef)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\032org.tensorflow.distruntimeB\rClusterProtosP\001ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\370\001\001'
+  _JOBDEF_TASKSENTRY._options = None
+  _JOBDEF_TASKSENTRY._serialized_options = b'8\001'
+  _JOBDEF._serialized_start=55
+  _JOBDEF._serialized_end=170
+  _JOBDEF_TASKSENTRY._serialized_start=126
+  _JOBDEF_TASKSENTRY._serialized_end=170
+  _CLUSTERDEF._serialized_start=172
+  _CLUSTERDEF._serialized_end=218
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/config_pb2.py

@@ -0,0 +1,271 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/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 coordination_config_pb2 as tensorbored_dot_compat_dot_proto_dot_coordination__config__pb2
+from tensorbored.compat.proto import cost_graph_pb2 as tensorbored_dot_compat_dot_proto_dot_cost__graph__pb2
+from tensorbored.compat.proto import graph_pb2 as tensorbored_dot_compat_dot_proto_dot_graph__pb2
+from tensorbored.compat.proto import step_stats_pb2 as tensorbored_dot_compat_dot_proto_dot_step__stats__pb2
+from tensorbored.compat.proto import cluster_pb2 as tensorbored_dot_compat_dot_proto_dot_cluster__pb2
+from tensorbored.compat.proto import debug_pb2 as tensorbored_dot_compat_dot_proto_dot_debug__pb2
+from tensorbored.compat.proto import rewriter_config_pb2 as tensorbored_dot_compat_dot_proto_dot_rewriter__config__pb2
+from tensorbored.compat.proto import rpc_options_pb2 as tensorbored_dot_compat_dot_proto_dot_rpc__options__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%tensorbored/compat/proto/config.proto\x12\x0btensorbored\x1a\x32tensorbored/compat/proto/coordination_config.proto\x1a)tensorbored/compat/proto/cost_graph.proto\x1a$tensorbored/compat/proto/graph.proto\x1a)tensorbored/compat/proto/step_stats.proto\x1a&tensorbored/compat/proto/cluster.proto\x1a$tensorbored/compat/proto/debug.proto\x1a.tensorbored/compat/proto/rewriter_config.proto\x1a*tensorbored/compat/proto/rpc_options.proto\"\x8c\n\n\nGPUOptions\x12\'\n\x1fper_process_gpu_memory_fraction\x18\x01 \x01(\x01\x12\x14\n\x0c\x61llow_growth\x18\x04 \x01(\x08\x12\x16\n\x0e\x61llocator_type\x18\x02 \x01(\t\x12\x1f\n\x17\x64\x65\x66\x65rred_deletion_bytes\x18\x03 \x01(\x03\x12\x1b\n\x13visible_device_list\x18\x05 \x01(\t\x12\"\n\x1apolling_active_delay_usecs\x18\x06 \x01(\x05\x12$\n\x1cpolling_inactive_delay_msecs\x18\x07 \x01(\x05\x12\x1c\n\x14\x66orce_gpu_compatible\x18\x08 \x01(\x08\x12:\n\x0c\x65xperimental\x18\t \x01(\x0b\x32$.tensorbored.GPUOptions.Experimental\x1a\xc4\x07\n\x0c\x45xperimental\x12L\n\x0fvirtual_devices\x18\x01 \x03(\x0b\x32\x33.tensorbored.GPUOptions.Experimental.VirtualDevices\x12#\n\x1bnum_virtual_devices_per_gpu\x18\x0f \x01(\x05\x12\x1a\n\x12use_unified_memory\x18\x02 \x01(\x08\x12#\n\x1bnum_dev_to_dev_copy_streams\x18\x03 \x01(\x05\x12\x1d\n\x15\x63ollective_ring_order\x18\x04 \x01(\t\x12\x1d\n\x15timestamped_allocator\x18\x05 \x01(\x08\x12#\n\x1bkernel_tracker_max_interval\x18\x07 \x01(\x05\x12 \n\x18kernel_tracker_max_bytes\x18\x08 \x01(\x05\x12\"\n\x1akernel_tracker_max_pending\x18\t \x01(\x05\x12\'\n\x1finternal_fragmentation_fraction\x18\n \x01(\x01\x12\x1d\n\x15use_cuda_malloc_async\x18\x0b \x01(\x08\x12,\n$disallow_retry_on_allocation_failure\x18\x0c \x01(\x08\x12 \n\x18gpu_host_mem_limit_in_mb\x18\r \x01(\x02\x12$\n\x1cgpu_host_mem_disallow_growth\x18\x0e \x01(\x08\x12$\n\x1cgpu_system_memory_size_in_mb\x18\x10 \x01(\x05\x12.\n&populate_pjrt_gpu_client_creation_info\x18\x11 \x01(\x08\x12\x0f\n\x07node_id\x18\x12 \x01(\x05\x12U\n\x14stream_merge_options\x18\x13 \x01(\x0b\x32\x37.tensorbored.GPUOptions.Experimental.StreamMergeOptions\x1aS\n\x0eVirtualDevices\x12\x17\n\x0fmemory_limit_mb\x18\x01 \x03(\x02\x12\x10\n\x08priority\x18\x02 \x03(\x05\x12\x16\n\x0e\x64\x65vice_ordinal\x18\x03 \x03(\x05\x1a\x85\x01\n\x12StreamMergeOptions\x12#\n\x1bmerge_host_to_device_stream\x18\x01 \x01(\x08\x12#\n\x1bmerge_device_to_host_stream\x18\x02 \x01(\x08\x12%\n\x1dmerge_device_to_device_stream\x18\x03 \x01(\x08\"\x9f\x03\n\x10OptimizerOptions\x12+\n#do_common_subexpression_elimination\x18\x01 \x01(\x08\x12\x1b\n\x13\x64o_constant_folding\x18\x02 \x01(\x08\x12$\n\x1cmax_folded_constant_in_bytes\x18\x06 \x01(\x03\x12\x1c\n\x14\x64o_function_inlining\x18\x04 \x01(\x08\x12\x36\n\topt_level\x18\x03 \x01(\x0e\x32#.tensorbored.OptimizerOptions.Level\x12\x46\n\x10global_jit_level\x18\x05 \x01(\x0e\x32,.tensorbored.OptimizerOptions.GlobalJitLevel\x12\x16\n\x0e\x63pu_global_jit\x18\x07 \x01(\x08\" \n\x05Level\x12\x06\n\x02L1\x10\x00\x12\x0f\n\x02L0\x10\xff\xff\xff\xff\xff\xff\xff\xff\xff\x01\"C\n\x0eGlobalJitLevel\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x10\n\x03OFF\x10\xff\xff\xff\xff\xff\xff\xff\xff\xff\x01\x12\x08\n\x04ON_1\x10\x01\x12\x08\n\x04ON_2\x10\x02\"\xf0\x02\n\x0cGraphOptions\x12\x1e\n\x16\x65nable_recv_scheduling\x18\x02 \x01(\x08\x12\x38\n\x11optimizer_options\x18\x03 \x01(\x0b\x32\x1d.tensorbored.OptimizerOptions\x12\x18\n\x10\x62uild_cost_model\x18\x04 \x01(\x03\x12\x1e\n\x16\x62uild_cost_model_after\x18\t \x01(\x03\x12\x14\n\x0cinfer_shapes\x18\x05 \x01(\x08\x12\x1a\n\x12place_pruned_graph\x18\x06 \x01(\x08\x12 \n\x18\x65nable_bfloat16_sendrecv\x18\x07 \x01(\x08\x12\x15\n\rtimeline_step\x18\x08 \x01(\x05\x12\x34\n\x0frewrite_options\x18\n \x01(\x0b\x32\x1b.tensorbored.RewriterConfigJ\x04\x08\x01\x10\x02R%skip_common_subexpression_elimination\"A\n\x15ThreadPoolOptionProto\x12\x13\n\x0bnum_threads\x18\x01 \x01(\x05\x12\x13\n\x0bglobal_name\x18\x02 \x01(\t\"0\n\x0fSessionMetadata\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\x03\"\xbf\x11\n\x0b\x43onfigProto\x12?\n\x0c\x64\x65vice_count\x18\x01 \x03(\x0b\x32).tensorbored.ConfigProto.DeviceCountEntry\x12$\n\x1cintra_op_parallelism_threads\x18\x02 \x01(\x05\x12$\n\x1cinter_op_parallelism_threads\x18\x05 \x01(\x05\x12\x1f\n\x17use_per_session_threads\x18\t \x01(\x08\x12H\n\x1csession_inter_op_thread_pool\x18\x0c \x03(\x0b\x32\".tensorbored.ThreadPoolOptionProto\x12\x18\n\x10placement_period\x18\x03 \x01(\x05\x12\x16\n\x0e\x64\x65vice_filters\x18\x04 \x03(\t\x12,\n\x0bgpu_options\x18\x06 \x01(\x0b\x32\x17.tensorbored.GPUOptions\x12\x39\n\x18pluggable_device_options\x18\x12 \x01(\x0b\x32\x17.tensorbored.GPUOptions\x12\x1c\n\x14\x61llow_soft_placement\x18\x07 \x01(\x08\x12\x1c\n\x14log_device_placement\x18\x08 \x01(\x08\x12\x30\n\rgraph_options\x18\n \x01(\x0b\x32\x19.tensorbored.GraphOptions\x12\x1f\n\x17operation_timeout_in_ms\x18\x0b \x01(\x03\x12,\n\x0brpc_options\x18\r \x01(\x0b\x32\x17.tensorbored.RPCOptions\x12,\n\x0b\x63luster_def\x18\x0e \x01(\x0b\x32\x17.tensorbored.ClusterDef\x12\x1d\n\x15isolate_session_state\x18\x0f \x01(\x08\x12(\n share_cluster_devices_in_session\x18\x11 \x01(\x08\x12;\n\x0c\x65xperimental\x18\x10 \x01(\x0b\x32%.tensorbored.ConfigProto.Experimental\x1a\x32\n\x10\x44\x65viceCountEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\x1a\xf7\n\n\x0c\x45xperimental\x12\x1f\n\x17\x63ollective_group_leader\x18\x01 \x01(\t\x12\x15\n\rexecutor_type\x18\x03 \x01(\t\x12\x1a\n\x12recv_buf_max_chunk\x18\x04 \x01(\x05\x12\x19\n\x11use_numa_affinity\x18\x05 \x01(\x08\x12\x35\n-collective_deterministic_sequential_execution\x18\x06 \x01(\x08\x12\x17\n\x0f\x63ollective_nccl\x18\x07 \x01(\x08\x12\x36\n.share_session_state_in_clusterspec_propagation\x18\x08 \x01(\x08\x12\x1f\n\x17\x64isable_thread_spinning\x18\t \x01(\x08\x12(\n share_cluster_devices_in_session\x18\n \x01(\x08\x12\x36\n\x10session_metadata\x18\x0b \x01(\x0b\x32\x1c.tensorbored.SessionMetadata\x12!\n\x19optimize_for_static_graph\x18\x0c \x01(\x08\x12\x1a\n\x12\x65nable_mlir_bridge\x18\r \x01(\x08\x12T\n\x13mlir_bridge_rollout\x18\x11 \x01(\x0e\x32\x37.tensorbored.ConfigProto.Experimental.MlirBridgeRollout\x12&\n\x1e\x65nable_mlir_graph_optimization\x18\x10 \x01(\x08\x12\'\n\x1f\x64isable_output_partition_graphs\x18\x0e \x01(\x08\x12#\n\x1bxla_fusion_autotuner_thresh\x18\x0f \x01(\x03\x12\x10\n\x08use_tfrt\x18\x12 \x01(\x08\x12\x19\n\x11\x65nable_multi_host\x18\x1b \x01(\x08\x12\x15\n\rtfrt_use_ifrt\x18  \x01(\x08\x12\x1b\n\x13\x62\x61\x63kend_server_port\x18\x1c \x01(\x05\x12\x12\n\ntarget_tpu\x18\x1d \x01(\x08\x12\x12\n\ntarget_gpu\x18\x1e \x01(\x08\x12\x1e\n\x16stream_merge_threshold\x18\x1f \x01(\x05\x12\'\n\x1f\x64isable_functional_ops_lowering\x18\x15 \x01(\x08\x12\'\n\x1fxla_prefer_single_graph_cluster\x18\x16 \x01(\x08\x12\x43\n\x13\x63oordination_config\x18\x17 \x01(\x0b\x32&.tensorbored.CoordinationServiceConfig\x12)\n!disable_optimize_for_static_graph\x18\x18 \x01(\x08\x12\x30\n(disable_eager_executor_streaming_enqueue\x18\x1a \x01(\x08\x12)\n!finalize_function_library_runtime\x18! \x01(\x08\x12!\n\x19\x66inalize_resource_manager\x18\" \x01(\x08\"\xde\x01\n\x11MlirBridgeRollout\x12#\n\x1fMLIR_BRIDGE_ROLLOUT_UNSPECIFIED\x10\x00\x12\x1f\n\x1bMLIR_BRIDGE_ROLLOUT_ENABLED\x10\x01\x12 \n\x1cMLIR_BRIDGE_ROLLOUT_DISABLED\x10\x02\"\x04\x08\x03\x10\x03\"\x04\x08\x04\x10\x04*%MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED*.MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLEDJ\x04\x08\x02\x10\x03J\x04\x08\x13\x10\x14J\x04\x08\x14\x10\x15J\x04\x08\x19\x10\x1a\"\xe5\x04\n\nRunOptions\x12\x37\n\x0btrace_level\x18\x01 \x01(\x0e\x32\".tensorbored.RunOptions.TraceLevel\x12\x15\n\rtimeout_in_ms\x18\x02 \x01(\x03\x12\x1c\n\x14inter_op_thread_pool\x18\x03 \x01(\x05\x12\x1f\n\x17output_partition_graphs\x18\x05 \x01(\x08\x12\x30\n\rdebug_options\x18\x06 \x01(\x0b\x32\x19.tensorbored.DebugOptions\x12*\n\"report_tensor_allocations_upon_oom\x18\x07 \x01(\x08\x12:\n\x0c\x65xperimental\x18\x08 \x01(\x0b\x32$.tensorbored.RunOptions.Experimental\x1a\xd3\x01\n\x0c\x45xperimental\x12\x1c\n\x14\x63ollective_graph_key\x18\x01 \x01(\x03\x12\x1c\n\x14use_run_handler_pool\x18\x02 \x01(\x08\x12\\\n\x18run_handler_pool_options\x18\x03 \x01(\x0b\x32:.tensorbored.RunOptions.Experimental.RunHandlerPoolOptions\x1a)\n\x15RunHandlerPoolOptions\x12\x10\n\x08priority\x18\x01 \x01(\x03\"R\n\nTraceLevel\x12\x0c\n\x08NO_TRACE\x10\x00\x12\x12\n\x0eSOFTWARE_TRACE\x10\x01\x12\x12\n\x0eHARDWARE_TRACE\x10\x02\x12\x0e\n\nFULL_TRACE\x10\x03J\x04\x08\x04\x10\x05\"\xc6\x03\n\x0bRunMetadata\x12*\n\nstep_stats\x18\x01 \x01(\x0b\x32\x16.tensorbored.StepStats\x12-\n\ncost_graph\x18\x02 \x01(\x0b\x32\x19.tensorbored.CostGraphDef\x12/\n\x10partition_graphs\x18\x03 \x03(\x0b\x32\x15.tensorbored.GraphDef\x12@\n\x0f\x66unction_graphs\x18\x04 \x03(\x0b\x32\'.tensorbored.RunMetadata.FunctionGraphs\x12\x36\n\x10session_metadata\x18\x05 \x01(\x0b\x32\x1c.tensorbored.SessionMetadata\x1a\xb0\x01\n\x0e\x46unctionGraphs\x12/\n\x10partition_graphs\x18\x01 \x03(\x0b\x32\x15.tensorbored.GraphDef\x12\x35\n\x16pre_optimization_graph\x18\x02 \x01(\x0b\x32\x15.tensorbored.GraphDef\x12\x36\n\x17post_optimization_graph\x18\x03 \x01(\x0b\x32\x15.tensorbored.GraphDef\":\n\x10TensorConnection\x12\x13\n\x0b\x66rom_tensor\x18\x01 \x01(\t\x12\x11\n\tto_tensor\x18\x02 \x01(\t\"\xb4\x03\n\x0f\x43\x61llableOptions\x12\x0c\n\x04\x66\x65\x65\x64\x18\x01 \x03(\t\x12\r\n\x05\x66\x65tch\x18\x02 \x03(\t\x12\x0e\n\x06target\x18\x03 \x03(\t\x12,\n\x0brun_options\x18\x04 \x01(\x0b\x32\x17.tensorbored.RunOptions\x12\x38\n\x11tensor_connection\x18\x05 \x03(\x0b\x32\x1d.tensorbored.TensorConnection\x12\x43\n\x0c\x66\x65\x65\x64_devices\x18\x06 \x03(\x0b\x32-.tensorbored.CallableOptions.FeedDevicesEntry\x12\x45\n\rfetch_devices\x18\x07 \x03(\x0b\x32..tensorbored.CallableOptions.FetchDevicesEntry\x12\x17\n\x0f\x66\x65tch_skip_sync\x18\x08 \x01(\x08\x1a\x32\n\x10\x46\x65\x65\x64\x44\x65vicesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\x33\n\x11\x46\x65tchDevicesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x9d\x01\n\x0f\x42\x61tchingOptions\x12\x19\n\x11num_batch_threads\x18\x01 \x01(\x05\x12\x16\n\x0emax_batch_size\x18\x02 \x01(\x05\x12\x1c\n\x14\x62\x61tch_timeout_micros\x18\x03 \x01(\x05\x12\x1b\n\x13\x61llowed_batch_sizes\x18\x04 \x03(\x05\x12\x1c\n\x14max_enqueued_batches\x18\x05 \x01(\x05\x42\x84\x01\n\x18org.tensorflow.frameworkB\x0c\x43onfigProtosP\x01ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_GPUOPTIONS = DESCRIPTOR.message_types_by_name['GPUOptions']
+_GPUOPTIONS_EXPERIMENTAL = _GPUOPTIONS.nested_types_by_name['Experimental']
+_GPUOPTIONS_EXPERIMENTAL_VIRTUALDEVICES = _GPUOPTIONS_EXPERIMENTAL.nested_types_by_name['VirtualDevices']
+_GPUOPTIONS_EXPERIMENTAL_STREAMMERGEOPTIONS = _GPUOPTIONS_EXPERIMENTAL.nested_types_by_name['StreamMergeOptions']
+_OPTIMIZEROPTIONS = DESCRIPTOR.message_types_by_name['OptimizerOptions']
+_GRAPHOPTIONS = DESCRIPTOR.message_types_by_name['GraphOptions']
+_THREADPOOLOPTIONPROTO = DESCRIPTOR.message_types_by_name['ThreadPoolOptionProto']
+_SESSIONMETADATA = DESCRIPTOR.message_types_by_name['SessionMetadata']
+_CONFIGPROTO = DESCRIPTOR.message_types_by_name['ConfigProto']
+_CONFIGPROTO_DEVICECOUNTENTRY = _CONFIGPROTO.nested_types_by_name['DeviceCountEntry']
+_CONFIGPROTO_EXPERIMENTAL = _CONFIGPROTO.nested_types_by_name['Experimental']
+_RUNOPTIONS = DESCRIPTOR.message_types_by_name['RunOptions']
+_RUNOPTIONS_EXPERIMENTAL = _RUNOPTIONS.nested_types_by_name['Experimental']
+_RUNOPTIONS_EXPERIMENTAL_RUNHANDLERPOOLOPTIONS = _RUNOPTIONS_EXPERIMENTAL.nested_types_by_name['RunHandlerPoolOptions']
+_RUNMETADATA = DESCRIPTOR.message_types_by_name['RunMetadata']
+_RUNMETADATA_FUNCTIONGRAPHS = _RUNMETADATA.nested_types_by_name['FunctionGraphs']
+_TENSORCONNECTION = DESCRIPTOR.message_types_by_name['TensorConnection']
+_CALLABLEOPTIONS = DESCRIPTOR.message_types_by_name['CallableOptions']
+_CALLABLEOPTIONS_FEEDDEVICESENTRY = _CALLABLEOPTIONS.nested_types_by_name['FeedDevicesEntry']
+_CALLABLEOPTIONS_FETCHDEVICESENTRY = _CALLABLEOPTIONS.nested_types_by_name['FetchDevicesEntry']
+_BATCHINGOPTIONS = DESCRIPTOR.message_types_by_name['BatchingOptions']
+_OPTIMIZEROPTIONS_LEVEL = _OPTIMIZEROPTIONS.enum_types_by_name['Level']
+_OPTIMIZEROPTIONS_GLOBALJITLEVEL = _OPTIMIZEROPTIONS.enum_types_by_name['GlobalJitLevel']
+_CONFIGPROTO_EXPERIMENTAL_MLIRBRIDGEROLLOUT = _CONFIGPROTO_EXPERIMENTAL.enum_types_by_name['MlirBridgeRollout']
+_RUNOPTIONS_TRACELEVEL = _RUNOPTIONS.enum_types_by_name['TraceLevel']
+GPUOptions = _reflection.GeneratedProtocolMessageType('GPUOptions', (_message.Message,), {
+
+  'Experimental' : _reflection.GeneratedProtocolMessageType('Experimental', (_message.Message,), {
+
+    'VirtualDevices' : _reflection.GeneratedProtocolMessageType('VirtualDevices', (_message.Message,), {
+      'DESCRIPTOR' : _GPUOPTIONS_EXPERIMENTAL_VIRTUALDEVICES,
+      '__module__' : 'tensorbored.compat.proto.config_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.GPUOptions.Experimental.VirtualDevices)
+      })
+    ,
+
+    'StreamMergeOptions' : _reflection.GeneratedProtocolMessageType('StreamMergeOptions', (_message.Message,), {
+      'DESCRIPTOR' : _GPUOPTIONS_EXPERIMENTAL_STREAMMERGEOPTIONS,
+      '__module__' : 'tensorbored.compat.proto.config_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.GPUOptions.Experimental.StreamMergeOptions)
+      })
+    ,
+    'DESCRIPTOR' : _GPUOPTIONS_EXPERIMENTAL,
+    '__module__' : 'tensorbored.compat.proto.config_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.GPUOptions.Experimental)
+    })
+  ,
+  'DESCRIPTOR' : _GPUOPTIONS,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.GPUOptions)
+  })
+_sym_db.RegisterMessage(GPUOptions)
+_sym_db.RegisterMessage(GPUOptions.Experimental)
+_sym_db.RegisterMessage(GPUOptions.Experimental.VirtualDevices)
+_sym_db.RegisterMessage(GPUOptions.Experimental.StreamMergeOptions)
+
+OptimizerOptions = _reflection.GeneratedProtocolMessageType('OptimizerOptions', (_message.Message,), {
+  'DESCRIPTOR' : _OPTIMIZEROPTIONS,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.OptimizerOptions)
+  })
+_sym_db.RegisterMessage(OptimizerOptions)
+
+GraphOptions = _reflection.GeneratedProtocolMessageType('GraphOptions', (_message.Message,), {
+  'DESCRIPTOR' : _GRAPHOPTIONS,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.GraphOptions)
+  })
+_sym_db.RegisterMessage(GraphOptions)
+
+ThreadPoolOptionProto = _reflection.GeneratedProtocolMessageType('ThreadPoolOptionProto', (_message.Message,), {
+  'DESCRIPTOR' : _THREADPOOLOPTIONPROTO,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ThreadPoolOptionProto)
+  })
+_sym_db.RegisterMessage(ThreadPoolOptionProto)
+
+SessionMetadata = _reflection.GeneratedProtocolMessageType('SessionMetadata', (_message.Message,), {
+  'DESCRIPTOR' : _SESSIONMETADATA,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SessionMetadata)
+  })
+_sym_db.RegisterMessage(SessionMetadata)
+
+ConfigProto = _reflection.GeneratedProtocolMessageType('ConfigProto', (_message.Message,), {
+
+  'DeviceCountEntry' : _reflection.GeneratedProtocolMessageType('DeviceCountEntry', (_message.Message,), {
+    'DESCRIPTOR' : _CONFIGPROTO_DEVICECOUNTENTRY,
+    '__module__' : 'tensorbored.compat.proto.config_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ConfigProto.DeviceCountEntry)
+    })
+  ,
+
+  'Experimental' : _reflection.GeneratedProtocolMessageType('Experimental', (_message.Message,), {
+    'DESCRIPTOR' : _CONFIGPROTO_EXPERIMENTAL,
+    '__module__' : 'tensorbored.compat.proto.config_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ConfigProto.Experimental)
+    })
+  ,
+  'DESCRIPTOR' : _CONFIGPROTO,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ConfigProto)
+  })
+_sym_db.RegisterMessage(ConfigProto)
+_sym_db.RegisterMessage(ConfigProto.DeviceCountEntry)
+_sym_db.RegisterMessage(ConfigProto.Experimental)
+
+RunOptions = _reflection.GeneratedProtocolMessageType('RunOptions', (_message.Message,), {
+
+  'Experimental' : _reflection.GeneratedProtocolMessageType('Experimental', (_message.Message,), {
+
+    'RunHandlerPoolOptions' : _reflection.GeneratedProtocolMessageType('RunHandlerPoolOptions', (_message.Message,), {
+      'DESCRIPTOR' : _RUNOPTIONS_EXPERIMENTAL_RUNHANDLERPOOLOPTIONS,
+      '__module__' : 'tensorbored.compat.proto.config_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.RunOptions.Experimental.RunHandlerPoolOptions)
+      })
+    ,
+    'DESCRIPTOR' : _RUNOPTIONS_EXPERIMENTAL,
+    '__module__' : 'tensorbored.compat.proto.config_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.RunOptions.Experimental)
+    })
+  ,
+  'DESCRIPTOR' : _RUNOPTIONS,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.RunOptions)
+  })
+_sym_db.RegisterMessage(RunOptions)
+_sym_db.RegisterMessage(RunOptions.Experimental)
+_sym_db.RegisterMessage(RunOptions.Experimental.RunHandlerPoolOptions)
+
+RunMetadata = _reflection.GeneratedProtocolMessageType('RunMetadata', (_message.Message,), {
+
+  'FunctionGraphs' : _reflection.GeneratedProtocolMessageType('FunctionGraphs', (_message.Message,), {
+    'DESCRIPTOR' : _RUNMETADATA_FUNCTIONGRAPHS,
+    '__module__' : 'tensorbored.compat.proto.config_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.RunMetadata.FunctionGraphs)
+    })
+  ,
+  'DESCRIPTOR' : _RUNMETADATA,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.RunMetadata)
+  })
+_sym_db.RegisterMessage(RunMetadata)
+_sym_db.RegisterMessage(RunMetadata.FunctionGraphs)
+
+TensorConnection = _reflection.GeneratedProtocolMessageType('TensorConnection', (_message.Message,), {
+  'DESCRIPTOR' : _TENSORCONNECTION,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.TensorConnection)
+  })
+_sym_db.RegisterMessage(TensorConnection)
+
+CallableOptions = _reflection.GeneratedProtocolMessageType('CallableOptions', (_message.Message,), {
+
+  'FeedDevicesEntry' : _reflection.GeneratedProtocolMessageType('FeedDevicesEntry', (_message.Message,), {
+    'DESCRIPTOR' : _CALLABLEOPTIONS_FEEDDEVICESENTRY,
+    '__module__' : 'tensorbored.compat.proto.config_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CallableOptions.FeedDevicesEntry)
+    })
+  ,
+
+  'FetchDevicesEntry' : _reflection.GeneratedProtocolMessageType('FetchDevicesEntry', (_message.Message,), {
+    'DESCRIPTOR' : _CALLABLEOPTIONS_FETCHDEVICESENTRY,
+    '__module__' : 'tensorbored.compat.proto.config_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CallableOptions.FetchDevicesEntry)
+    })
+  ,
+  'DESCRIPTOR' : _CALLABLEOPTIONS,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.CallableOptions)
+  })
+_sym_db.RegisterMessage(CallableOptions)
+_sym_db.RegisterMessage(CallableOptions.FeedDevicesEntry)
+_sym_db.RegisterMessage(CallableOptions.FetchDevicesEntry)
+
+BatchingOptions = _reflection.GeneratedProtocolMessageType('BatchingOptions', (_message.Message,), {
+  'DESCRIPTOR' : _BATCHINGOPTIONS,
+  '__module__' : 'tensorbored.compat.proto.config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.BatchingOptions)
+  })
+_sym_db.RegisterMessage(BatchingOptions)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\014ConfigProtosP\001ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\370\001\001'
+  _CONFIGPROTO_DEVICECOUNTENTRY._options = None
+  _CONFIGPROTO_DEVICECOUNTENTRY._serialized_options = b'8\001'
+  _CALLABLEOPTIONS_FEEDDEVICESENTRY._options = None
+  _CALLABLEOPTIONS_FEEDDEVICESENTRY._serialized_options = b'8\001'
+  _CALLABLEOPTIONS_FETCHDEVICESENTRY._options = None
+  _CALLABLEOPTIONS_FETCHDEVICESENTRY._serialized_options = b'8\001'
+  _GPUOPTIONS._serialized_start=401
+  _GPUOPTIONS._serialized_end=1693
+  _GPUOPTIONS_EXPERIMENTAL._serialized_start=729
+  _GPUOPTIONS_EXPERIMENTAL._serialized_end=1693
+  _GPUOPTIONS_EXPERIMENTAL_VIRTUALDEVICES._serialized_start=1474
+  _GPUOPTIONS_EXPERIMENTAL_VIRTUALDEVICES._serialized_end=1557
+  _GPUOPTIONS_EXPERIMENTAL_STREAMMERGEOPTIONS._serialized_start=1560
+  _GPUOPTIONS_EXPERIMENTAL_STREAMMERGEOPTIONS._serialized_end=1693
+  _OPTIMIZEROPTIONS._serialized_start=1696
+  _OPTIMIZEROPTIONS._serialized_end=2111
+  _OPTIMIZEROPTIONS_LEVEL._serialized_start=2010
+  _OPTIMIZEROPTIONS_LEVEL._serialized_end=2042
+  _OPTIMIZEROPTIONS_GLOBALJITLEVEL._serialized_start=2044
+  _OPTIMIZEROPTIONS_GLOBALJITLEVEL._serialized_end=2111
+  _GRAPHOPTIONS._serialized_start=2114
+  _GRAPHOPTIONS._serialized_end=2482
+  _THREADPOOLOPTIONPROTO._serialized_start=2484
+  _THREADPOOLOPTIONPROTO._serialized_end=2549
+  _SESSIONMETADATA._serialized_start=2551
+  _SESSIONMETADATA._serialized_end=2599
+  _CONFIGPROTO._serialized_start=2602
+  _CONFIGPROTO._serialized_end=4841
+  _CONFIGPROTO_DEVICECOUNTENTRY._serialized_start=3389
+  _CONFIGPROTO_DEVICECOUNTENTRY._serialized_end=3439
+  _CONFIGPROTO_EXPERIMENTAL._serialized_start=3442
+  _CONFIGPROTO_EXPERIMENTAL._serialized_end=4841
+  _CONFIGPROTO_EXPERIMENTAL_MLIRBRIDGEROLLOUT._serialized_start=4595
+  _CONFIGPROTO_EXPERIMENTAL_MLIRBRIDGEROLLOUT._serialized_end=4817
+  _RUNOPTIONS._serialized_start=4844
+  _RUNOPTIONS._serialized_end=5457
+  _RUNOPTIONS_EXPERIMENTAL._serialized_start=5156
+  _RUNOPTIONS_EXPERIMENTAL._serialized_end=5367
+  _RUNOPTIONS_EXPERIMENTAL_RUNHANDLERPOOLOPTIONS._serialized_start=5326
+  _RUNOPTIONS_EXPERIMENTAL_RUNHANDLERPOOLOPTIONS._serialized_end=5367
+  _RUNOPTIONS_TRACELEVEL._serialized_start=5369
+  _RUNOPTIONS_TRACELEVEL._serialized_end=5451
+  _RUNMETADATA._serialized_start=5460
+  _RUNMETADATA._serialized_end=5914
+  _RUNMETADATA_FUNCTIONGRAPHS._serialized_start=5738
+  _RUNMETADATA_FUNCTIONGRAPHS._serialized_end=5914
+  _TENSORCONNECTION._serialized_start=5916
+  _TENSORCONNECTION._serialized_end=5974
+  _CALLABLEOPTIONS._serialized_start=5977
+  _CALLABLEOPTIONS._serialized_end=6413
+  _CALLABLEOPTIONS_FEEDDEVICESENTRY._serialized_start=6310
+  _CALLABLEOPTIONS_FEEDDEVICESENTRY._serialized_end=6360
+  _CALLABLEOPTIONS_FETCHDEVICESENTRY._serialized_start=6362
+  _CALLABLEOPTIONS_FETCHDEVICESENTRY._serialized_end=6413
+  _BATCHINGOPTIONS._serialized_start=6416
+  _BATCHINGOPTIONS._serialized_end=6573
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/coordination_config_pb2.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/coordination_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()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n2tensorbored/compat/proto/coordination_config.proto\x12\x0btensorbored\"1\n\x0e\x43oordinatedJob\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\tnum_tasks\x18\x02 \x01(\x05\"\xf8\x03\n\x19\x43oordinationServiceConfig\x12\x14\n\x0cservice_type\x18\x01 \x01(\t\x12\x16\n\x0eservice_leader\x18\x02 \x01(\t\x12\x1b\n\x13\x65nable_health_check\x18\x03 \x01(\x08\x12&\n\x1e\x63luster_register_timeout_in_ms\x18\x04 \x01(\x03\x12%\n\x1d\x63luster_register_with_barrier\x18\x0e \x01(\x08\x12\x1f\n\x17heartbeat_timeout_in_ms\x18\x05 \x01(\x03\x12\x39\n\x14\x63oordinated_job_list\x18\n \x03(\x0b\x32\x1b.tensorbored.CoordinatedJob\x12&\n\x1eshutdown_barrier_timeout_in_ms\x18\x07 \x01(\x03\x12*\n\"agent_destruction_without_shutdown\x18\x08 \x01(\x08\x12\x18\n\x10recoverable_jobs\x18\t \x03(\t\x12*\n\"allow_new_incarnation_to_reconnect\x18\x0b \x01(\x08\x12\x15\n\rforce_disable\x18\x0c \x01(\x08\x12.\n&poll_for_error_from_service_at_startup\x18\r \x01(\x08J\x04\x08\x06\x10\x07\x42WZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_protob\x06proto3')
+
+
+
+_COORDINATEDJOB = DESCRIPTOR.message_types_by_name['CoordinatedJob']
+_COORDINATIONSERVICECONFIG = DESCRIPTOR.message_types_by_name['CoordinationServiceConfig']
+CoordinatedJob = _reflection.GeneratedProtocolMessageType('CoordinatedJob', (_message.Message,), {
+  'DESCRIPTOR' : _COORDINATEDJOB,
+  '__module__' : 'tensorbored.compat.proto.coordination_config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.CoordinatedJob)
+  })
+_sym_db.RegisterMessage(CoordinatedJob)
+
+CoordinationServiceConfig = _reflection.GeneratedProtocolMessageType('CoordinationServiceConfig', (_message.Message,), {
+  'DESCRIPTOR' : _COORDINATIONSERVICECONFIG,
+  '__module__' : 'tensorbored.compat.proto.coordination_config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.CoordinationServiceConfig)
+  })
+_sym_db.RegisterMessage(CoordinationServiceConfig)
+
+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'
+  _COORDINATEDJOB._serialized_start=67
+  _COORDINATEDJOB._serialized_end=116
+  _COORDINATIONSERVICECONFIG._serialized_start=119
+  _COORDINATIONSERVICECONFIG._serialized_end=623
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/cost_graph_pb2.py

@@ -0,0 +1,87 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/cost_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 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/cost_graph.proto\x12\x0btensorbored\x1a+tensorbored/compat/proto/tensor_shape.proto\x1a$tensorbored/compat/proto/types.proto\"\xd0\x06\n\x0c\x43ostGraphDef\x12,\n\x04node\x18\x01 \x03(\x0b\x32\x1e.tensorbored.CostGraphDef.Node\x12\x36\n\x04\x63ost\x18\x02 \x03(\x0b\x32(.tensorbored.CostGraphDef.AggregatedCost\x1a\xa6\x05\n\x04Node\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0e\n\x06\x64\x65vice\x18\x02 \x01(\t\x12\n\n\x02id\x18\x03 \x01(\x05\x12<\n\ninput_info\x18\x04 \x03(\x0b\x32(.tensorbored.CostGraphDef.Node.InputInfo\x12>\n\x0boutput_info\x18\x05 \x03(\x0b\x32).tensorbored.CostGraphDef.Node.OutputInfo\x12\x1d\n\x15temporary_memory_size\x18\x06 \x01(\x03\x12\x1e\n\x16persistent_memory_size\x18\x0c \x01(\x03\x12!\n\x15host_temp_memory_size\x18\n \x01(\x03\x42\x02\x18\x01\x12#\n\x17\x64\x65vice_temp_memory_size\x18\x0b \x01(\x03\x42\x02\x18\x01\x12)\n\x1d\x64\x65vice_persistent_memory_size\x18\x10 \x01(\x03\x42\x02\x18\x01\x12\x14\n\x0c\x63ompute_cost\x18\t \x01(\x03\x12\x14\n\x0c\x63ompute_time\x18\x0e \x01(\x03\x12\x13\n\x0bmemory_time\x18\x0f \x01(\x03\x12\x10\n\x08is_final\x18\x07 \x01(\x08\x12\x15\n\rcontrol_input\x18\x08 \x03(\x05\x12\x12\n\ninaccurate\x18\x11 \x01(\x08\x1a;\n\tInputInfo\x12\x16\n\x0epreceding_node\x18\x01 \x01(\x05\x12\x16\n\x0epreceding_port\x18\x02 \x01(\x05\x1a\x88\x01\n\nOutputInfo\x12\x0c\n\x04size\x18\x01 \x01(\x03\x12\x18\n\x10\x61lias_input_port\x18\x02 \x01(\x03\x12,\n\x05shape\x18\x03 \x01(\x0b\x32\x1d.tensorbored.TensorShapeProto\x12$\n\x05\x64type\x18\x04 \x01(\x0e\x32\x15.tensorbored.DataType\x1a\x31\n\x0e\x41ggregatedCost\x12\x0c\n\x04\x63ost\x18\x01 \x01(\x02\x12\x11\n\tdimension\x18\x02 \x01(\tB\x83\x01\n\x18org.tensorflow.frameworkB\x0f\x43ostGraphProtosP\x01ZQgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/cost_graph_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_COSTGRAPHDEF = DESCRIPTOR.message_types_by_name['CostGraphDef']
+_COSTGRAPHDEF_NODE = _COSTGRAPHDEF.nested_types_by_name['Node']
+_COSTGRAPHDEF_NODE_INPUTINFO = _COSTGRAPHDEF_NODE.nested_types_by_name['InputInfo']
+_COSTGRAPHDEF_NODE_OUTPUTINFO = _COSTGRAPHDEF_NODE.nested_types_by_name['OutputInfo']
+_COSTGRAPHDEF_AGGREGATEDCOST = _COSTGRAPHDEF.nested_types_by_name['AggregatedCost']
+CostGraphDef = _reflection.GeneratedProtocolMessageType('CostGraphDef', (_message.Message,), {
+
+  'Node' : _reflection.GeneratedProtocolMessageType('Node', (_message.Message,), {
+
+    'InputInfo' : _reflection.GeneratedProtocolMessageType('InputInfo', (_message.Message,), {
+      'DESCRIPTOR' : _COSTGRAPHDEF_NODE_INPUTINFO,
+      '__module__' : 'tensorbored.compat.proto.cost_graph_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.CostGraphDef.Node.InputInfo)
+      })
+    ,
+
+    'OutputInfo' : _reflection.GeneratedProtocolMessageType('OutputInfo', (_message.Message,), {
+      'DESCRIPTOR' : _COSTGRAPHDEF_NODE_OUTPUTINFO,
+      '__module__' : 'tensorbored.compat.proto.cost_graph_pb2'
+      # @@protoc_insertion_point(class_scope:tensorbored.CostGraphDef.Node.OutputInfo)
+      })
+    ,
+    'DESCRIPTOR' : _COSTGRAPHDEF_NODE,
+    '__module__' : 'tensorbored.compat.proto.cost_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CostGraphDef.Node)
+    })
+  ,
+
+  'AggregatedCost' : _reflection.GeneratedProtocolMessageType('AggregatedCost', (_message.Message,), {
+    'DESCRIPTOR' : _COSTGRAPHDEF_AGGREGATEDCOST,
+    '__module__' : 'tensorbored.compat.proto.cost_graph_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CostGraphDef.AggregatedCost)
+    })
+  ,
+  'DESCRIPTOR' : _COSTGRAPHDEF,
+  '__module__' : 'tensorbored.compat.proto.cost_graph_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.CostGraphDef)
+  })
+_sym_db.RegisterMessage(CostGraphDef)
+_sym_db.RegisterMessage(CostGraphDef.Node)
+_sym_db.RegisterMessage(CostGraphDef.Node.InputInfo)
+_sym_db.RegisterMessage(CostGraphDef.Node.OutputInfo)
+_sym_db.RegisterMessage(CostGraphDef.AggregatedCost)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\017CostGraphProtosP\001ZQgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/cost_graph_go_proto\370\001\001'
+  _COSTGRAPHDEF_NODE.fields_by_name['host_temp_memory_size']._options = None
+  _COSTGRAPHDEF_NODE.fields_by_name['host_temp_memory_size']._serialized_options = b'\030\001'
+  _COSTGRAPHDEF_NODE.fields_by_name['device_temp_memory_size']._options = None
+  _COSTGRAPHDEF_NODE.fields_by_name['device_temp_memory_size']._serialized_options = b'\030\001'
+  _COSTGRAPHDEF_NODE.fields_by_name['device_persistent_memory_size']._options = None
+  _COSTGRAPHDEF_NODE.fields_by_name['device_persistent_memory_size']._serialized_options = b'\030\001'
+  _COSTGRAPHDEF._serialized_start=142
+  _COSTGRAPHDEF._serialized_end=990
+  _COSTGRAPHDEF_NODE._serialized_start=261
+  _COSTGRAPHDEF_NODE._serialized_end=939
+  _COSTGRAPHDEF_NODE_INPUTINFO._serialized_start=741
+  _COSTGRAPHDEF_NODE_INPUTINFO._serialized_end=800
+  _COSTGRAPHDEF_NODE_OUTPUTINFO._serialized_start=803
+  _COSTGRAPHDEF_NODE_OUTPUTINFO._serialized_end=939
+  _COSTGRAPHDEF_AGGREGATEDCOST._serialized_start=941
+  _COSTGRAPHDEF_AGGREGATEDCOST._serialized_end=990
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/cpp_shape_inference_pb2.py

@@ -0,0 +1,70 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/cpp_shape_inference.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 full_type_pb2 as tensorbored_dot_compat_dot_proto_dot_full__type__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'\n2tensorbored/compat/proto/cpp_shape_inference.proto\x12\x0btensorbored\x1a(tensorbored/compat/proto/full_type.proto\x1a+tensorbored/compat/proto/tensor_shape.proto\x1a$tensorbored/compat/proto/types.proto\"\xa1\x03\n\x17\x43ppShapeInferenceResult\x12,\n\x05shape\x18\x01 \x01(\x0b\x32\x1d.tensorbored.TensorShapeProto\x12\x44\n\x0bhandle_data\x18\x04 \x01(\x0b\x32/.tensorbored.CppShapeInferenceResult.HandleData\x1a\x96\x01\n\x12HandleShapeAndType\x12,\n\x05shape\x18\x01 \x01(\x0b\x32\x1d.tensorbored.TensorShapeProto\x12$\n\x05\x64type\x18\x02 \x01(\x0e\x32\x15.tensorbored.DataType\x12&\n\x04type\x18\x04 \x01(\x0b\x32\x18.tensorbored.FullTypeDefJ\x04\x08\x03\x10\x04\x1am\n\nHandleData\x12\x0e\n\x06is_set\x18\x01 \x01(\x08\x12O\n\x0eshape_and_type\x18\x02 \x03(\x0b\x32\x37.tensorbored.CppShapeInferenceResult.HandleShapeAndTypeJ\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04\"e\n\x1d\x43ppShapeInferenceInputsNeeded\x12\x1c\n\x14input_tensors_needed\x18\x01 \x03(\x05\x12&\n\x1einput_tensors_as_shapes_needed\x18\x02 \x03(\x05\x42\x61Z\\github.com/tensorflow/tensorflow/tensorflow/go/python/framework/cpp_shape_inference_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_CPPSHAPEINFERENCERESULT = DESCRIPTOR.message_types_by_name['CppShapeInferenceResult']
+_CPPSHAPEINFERENCERESULT_HANDLESHAPEANDTYPE = _CPPSHAPEINFERENCERESULT.nested_types_by_name['HandleShapeAndType']
+_CPPSHAPEINFERENCERESULT_HANDLEDATA = _CPPSHAPEINFERENCERESULT.nested_types_by_name['HandleData']
+_CPPSHAPEINFERENCEINPUTSNEEDED = DESCRIPTOR.message_types_by_name['CppShapeInferenceInputsNeeded']
+CppShapeInferenceResult = _reflection.GeneratedProtocolMessageType('CppShapeInferenceResult', (_message.Message,), {
+
+  'HandleShapeAndType' : _reflection.GeneratedProtocolMessageType('HandleShapeAndType', (_message.Message,), {
+    'DESCRIPTOR' : _CPPSHAPEINFERENCERESULT_HANDLESHAPEANDTYPE,
+    '__module__' : 'tensorbored.compat.proto.cpp_shape_inference_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CppShapeInferenceResult.HandleShapeAndType)
+    })
+  ,
+
+  'HandleData' : _reflection.GeneratedProtocolMessageType('HandleData', (_message.Message,), {
+    'DESCRIPTOR' : _CPPSHAPEINFERENCERESULT_HANDLEDATA,
+    '__module__' : 'tensorbored.compat.proto.cpp_shape_inference_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.CppShapeInferenceResult.HandleData)
+    })
+  ,
+  'DESCRIPTOR' : _CPPSHAPEINFERENCERESULT,
+  '__module__' : 'tensorbored.compat.proto.cpp_shape_inference_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.CppShapeInferenceResult)
+  })
+_sym_db.RegisterMessage(CppShapeInferenceResult)
+_sym_db.RegisterMessage(CppShapeInferenceResult.HandleShapeAndType)
+_sym_db.RegisterMessage(CppShapeInferenceResult.HandleData)
+
+CppShapeInferenceInputsNeeded = _reflection.GeneratedProtocolMessageType('CppShapeInferenceInputsNeeded', (_message.Message,), {
+  'DESCRIPTOR' : _CPPSHAPEINFERENCEINPUTSNEEDED,
+  '__module__' : 'tensorbored.compat.proto.cpp_shape_inference_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.CppShapeInferenceInputsNeeded)
+  })
+_sym_db.RegisterMessage(CppShapeInferenceInputsNeeded)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'Z\\github.com/tensorflow/tensorflow/tensorflow/go/python/framework/cpp_shape_inference_go_proto\370\001\001'
+  _CPPSHAPEINFERENCERESULT._serialized_start=193
+  _CPPSHAPEINFERENCERESULT._serialized_end=610
+  _CPPSHAPEINFERENCERESULT_HANDLESHAPEANDTYPE._serialized_start=337
+  _CPPSHAPEINFERENCERESULT_HANDLESHAPEANDTYPE._serialized_end=487
+  _CPPSHAPEINFERENCERESULT_HANDLEDATA._serialized_start=489
+  _CPPSHAPEINFERENCERESULT_HANDLEDATA._serialized_end=598
+  _CPPSHAPEINFERENCEINPUTSNEEDED._serialized_start=612
+  _CPPSHAPEINFERENCEINPUTSNEEDED._serialized_end=713
+# @@protoc_insertion_point(module_scope)