diff options
author | Sourabh Bajaj <sourabhbajaj@google.com> | 2018-09-06 10:50:35 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-06 10:55:17 -0700 |
commit | 9638524520d582e93a8038a89cd5cc62d719a3b6 (patch) | |
tree | e16b7b2c5032b15ac30ea07cbc36c199d6727272 /tensorflow/contrib/cluster_resolver | |
parent | b9310932ce2120c8c36eb69bc135748fd3caf897 (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.py | 4 |
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. |