aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/__init__.py
diff options
context:
space:
mode:
authorGravatar Brennan Saeta <saeta@google.com>2017-05-04 19:43:48 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-05-04 21:08:47 -0700
commitf28935a7d280b6ba75fe93fe35783d87b9cc2ec9 (patch)
treedb03a72f0dd29e8e09dcf7805684268b3d40a54c /tensorflow/python/__init__.py
parentefa08d80a53a95ce6b8beb61ec86a275aed6b6c7 (diff)
Implement ClusterSpec Propagation in TF Master
ClusterSpec propagation is a capability upgrade for TensorFlow that should make it much easier to (1) build distributed TensorFlow clusters, and (2) handle node failures. The ClusterSpec propagation capability allows TensorFlow workers to be booted independently of each other, and with no knowledge about others. The client can then construct a ClusterDef (ClusterSpec), and then send it to the TF master at session creation. The master in turn then propagates the ClusterDef along to all of the workers. Change: 155159972
Diffstat (limited to 'tensorflow/python/__init__.py')
-rw-r--r--tensorflow/python/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/__init__.py b/tensorflow/python/__init__.py
index 864a96ef34..6336ca2310 100644
--- a/tensorflow/python/__init__.py
+++ b/tensorflow/python/__init__.py
@@ -55,6 +55,7 @@ from tensorflow.core.framework.summary_pb2 import *
from tensorflow.core.framework.attr_value_pb2 import *
from tensorflow.core.protobuf.meta_graph_pb2 import TensorInfo
from tensorflow.core.protobuf.config_pb2 import *
+from tensorflow.core.protobuf.tensorflow_server_pb2 import *
from tensorflow.core.protobuf.rewriter_config_pb2 import *
from tensorflow.core.util.event_pb2 import *
@@ -131,6 +132,7 @@ _allowed_symbols = [
'AttrValue',
'AutoParallelOptions',
'ConfigProto',
+ 'ClusterDef',
'DeviceSpec',
'Event',
'GPUOptions',