experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Disable discovery cache for TPUClusterResolver.

Browse Source 
PiperOrigin-RevId: 246176508
This commit is contained in:
Michael Banfield 2019-05-01 12:21:56 -07:00 committed by TensorFlower Gardener
parent 014d9bcd23
commit 750ceb3938
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/distribute/cluster_resolver/tpu_cluster_resolver.py
View File

@ -108,13 +108,11 @@ class TPUClusterResolver(ClusterResolver):
if self._discovery_url: if self._discovery_url:
return discovery.build( return discovery.build(
'tpu', 'v1alpha1', 'tpu', 'v1alpha1', credentials=credentials,
credentials=credentials, discoveryServiceUrl=self._discovery_url, cache_discovery=False)
discoveryServiceUrl=self._discovery_url)
else: else:
return discovery.build( return discovery.build(
'tpu', 'v1alpha1', 'tpu', 'v1alpha1', credentials=credentials, cache_discovery=False)
credentials=credentials)
def _request_compute_metadata(self, path): def _request_compute_metadata(self, path):
req = Request('%s/computeMetadata/v1/%s' % (_GCE_METADATA_ENDPOINT, path), req = Request('%s/computeMetadata/v1/%s' % (_GCE_METADATA_ENDPOINT, path),