aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/cluster_resolver
diff options
context:
space:
mode:
authorGravatar Sourabh Bajaj <sourabhbajaj@google.com>2018-09-06 10:50:35 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-06 10:55:17 -0700
commit9638524520d582e93a8038a89cd5cc62d719a3b6 (patch)
treee16b7b2c5032b15ac30ea07cbc36c199d6727272 /tensorflow/contrib/cluster_resolver
parentb9310932ce2120c8c36eb69bc135748fd3caf897 (diff)
Job name should be picked based on the cluster_spec
PiperOrigin-RevId: 211833041
Diffstat (limited to 'tensorflow/contrib/cluster_resolver')
-rw-r--r--tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py b/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py
index 1ab150d74a..1056894f18 100644
--- a/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py
+++ b/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py
@@ -229,6 +229,10 @@ class TPUClusterResolver(ClusterResolver):
def get_master(self):
return self.master()
+ def get_job_name(self):
+ if self._shouldResolve():
+ return self._job_name
+
def cluster_spec(self):
"""Returns a ClusterSpec object based on the latest TPU information.