diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-04 03:20:59 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-04 03:28:32 -0700 |
| commit | 9cd6cab4f85f1f35c6532da3fb68839294d44ee4 (patch) | |
| tree | 987464be89615a5efb5bfcc4794e017b590decc8 /tensorflow/contrib/cluster_resolver | |
| parent | 6cc738da1748e819b9c8ee92dc2f1a7bdb291b50 (diff) | |
Internal change.
PiperOrigin-RevId: 215711454
Diffstat (limited to 'tensorflow/contrib/cluster_resolver')
| -rw-r--r-- | tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py | 1 |
1 files changed, 1 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 1056894f18..f4a8e16c99 100644 --- a/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py +++ b/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py @@ -60,6 +60,7 @@ class TPUClusterResolver(ClusterResolver): if (self._tpu == compat.as_bytes('') or self._tpu == compat.as_bytes('local') or self._tpu.startswith(compat.as_bytes('/bns')) or + self._tpu.startswith(compat.as_bytes('localhost:')) or self._tpu.startswith(compat.as_bytes('grpc://'))): return False return True |