Home
last modified time | relevance | path

Searched refs:is_running_in_gce (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dpreempted_hook.py43 if tpu_cluster_resolver.is_running_in_gce():
80 if not tpu_cluster_resolver.is_running_in_gce():
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dtpu_cluster_resolver.py22 ….python.distribute.cluster_resolver.tpu.tpu_cluster_resolver import is_running_in_gce # pylint: d…
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
Dtpu_cluster_resolver.py41 def is_running_in_gce(): function
Dtpu_cluster_resolver_test.py150 self.assertTrue(resolver.is_running_in_gce())