Home
last modified time | relevance | path

Searched refs:tpu_cluster_resolver (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/profiler/
Dcapture_tpu_profile.py27 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver as resolver
151 tpu_cluster_resolver = None
159 tpu_cluster_resolver = (
163 service_addr = tpu_cluster_resolver.get_master()
173 elif tpu_cluster_resolver is not None:
174 workers_list = get_workers_list(tpu_cluster_resolver)
/external/tensorflow/tensorflow/python/tpu/
Dpreempted_hook.py26 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
43 if tpu_cluster_resolver.is_running_in_gce():
80 if not tpu_cluster_resolver.is_running_in_gce():
Dasync_checkpoint_test.py26 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
100 resolver = tpu_cluster_resolver.TPUClusterResolver(
149 resolver = tpu_cluster_resolver.TPUClusterResolver(
Dtpu_strategy_util.py23 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
Dtpu_embedding_v2_correctness_test.py32 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
128 self.resolver = tpu_cluster_resolver.TPUClusterResolver(
Dtpu_outside_compilation_test.py31 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
64 resolver = tpu_cluster_resolver.TPUClusterResolver(
Dtpu_embedding_v2_test.py33 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
74 self.resolver = tpu_cluster_resolver.TPUClusterResolver(
506 self.resolver = tpu_cluster_resolver.TPUClusterResolver(
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dtpu_cluster_resolver.py22 from tensorflow.python.distribute.cluster_resolver.tpu.tpu_cluster_resolver import is_running_in_gc…
23 from tensorflow.python.distribute.cluster_resolver.tpu.tpu_cluster_resolver import TPUClusterResolv…
D__init__.py35 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
DBUILD72 srcs = ["tpu_cluster_resolver.py"],
/external/tensorflow/tensorflow/python/keras/distribute/
Dtpu_strategy_test_utils.py25 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
37 resolver = tpu_cluster_resolver.TPUClusterResolver(
/external/tensorflow/tensorflow/python/eager/
Dremote_cloud_tpu_test.py24 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
66 resolver = tpu_cluster_resolver.TPUClusterResolver(
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_compilation_test.py22 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
37 resolver = tpu_cluster_resolver.TPUClusterResolver(
Dtpu_strategy.py320 tpu_cluster_resolver=None, argument
334 self, tpu_cluster_resolver,
626 tpu_cluster_resolver=None, argument
641 self, tpu_cluster_resolver, device_assignment=device_assignment))
685 tpu_cluster_resolver=None, argument
703 self, tpu_cluster_resolver, steps_per_run, device_assignment))
788 tpu_cluster_resolver=None, argument
793 if tpu_cluster_resolver is None:
794 tpu_cluster_resolver = TPUClusterResolver("")
806 self._tpu_cluster_resolver = tpu_cluster_resolver
Dstrategy_combinations.py33 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
85 resolver = tpu_cluster_resolver.TPUClusterResolver()
91 resolver = tpu_cluster_resolver.TPUClusterResolver(
Dtpu_strategy_test.py33 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
72 resolver = tpu_cluster_resolver.TPUClusterResolver(
Dvars_test.py32 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
912 resolver = tpu_cluster_resolver.TPUClusterResolver("")
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
DBUILD18 srcs = ["tpu_cluster_resolver.py"],
Dtpu_cluster_resolver_test.py28 from tensorflow.python.distribute.cluster_resolver.tpu import tpu_cluster_resolver as resolver
/external/tensorflow/tensorflow/python/keras/tests/
Dautomatic_outside_compilation_test.py32 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
63 resolver = tpu_cluster_resolver.TPUClusterResolver(
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.tpu.tpu_cluster_resolver.TPUC…
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt21 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'device_assignment\'], varargs=None, keywords…
Dtensorflow.distribute.-t-p-u-strategy.pbtxt21 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'experimental_device_assignment\'], varargs=N…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.tpu.tpu_cluster_resolver.TPUC…
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt25 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'steps_per_run\', \'device_assignment\'], var…