| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/profiler/ |
| H A D | capture_tpu_profile.py | 24 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver as resolver
148 tpu_cluster_resolver = None
156 tpu_cluster_resolver = (
160 service_addr = tpu_cluster_resolver.get_master()
170 elif tpu_cluster_resolver is not None:
171 workers_list = get_workers_list(tpu_cluster_resolver)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | preempted_hook.py | 22 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
39 if tpu_cluster_resolver.is_running_in_gce():
76 if not tpu_cluster_resolver.is_running_in_gce():
|
| H A D | async_checkpoint_test.py | 22 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
96 resolver = tpu_cluster_resolver.TPUClusterResolver(
145 resolver = tpu_cluster_resolver.TPUClusterResolver(
|
| H A D | tpu_strategy_util.py | 23 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
|
| H A D | tpu_outside_compilation_test.py | 28 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
62 resolver = tpu_cluster_resolver.TPUClusterResolver(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | tpu_strategy.py | 333 tpu_cluster_resolver=None, argument
356 tpu_cluster_resolver,
681 tpu_cluster_resolver=None, argument
697 self, tpu_cluster_resolver, device_assignment=device_assignment))
741 tpu_cluster_resolver=None, argument
759 self, tpu_cluster_resolver, steps_per_run, device_assignment))
844 tpu_cluster_resolver=None, argument
850 if tpu_cluster_resolver is None:
851 tpu_cluster_resolver = TPUClusterResolver("")
863 self._tpu_cluster_resolver = tpu_cluster_resolver
|
| H A D | tpu_strategy_compilation_test.py | 18 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
33 resolver = tpu_cluster_resolver.TPUClusterResolver(
|
| H A D | strategy_combinations.py | 34 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
90 resolver = tpu_cluster_resolver.TPUClusterResolver()
96 resolver = tpu_cluster_resolver.TPUClusterResolver(
|
| H A D | tpu_strategy_model_parallelism_test.py | 25 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
53 resolver = tpu_cluster_resolver.TPUClusterResolver(
|
| H A D | tpu_strategy_test.py | 28 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
73 resolver = tpu_cluster_resolver.TPUClusterResolver(
|
| H A D | vars_test.py | 30 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
971 resolver = tpu_cluster_resolver.TPUClusterResolver("")
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| H A D | tpu_cluster_resolver.py | 18 from tensorflow.python.distribute.cluster_resolver.tpu.tpu_cluster_resolver import is_running_in_gc…
19 from tensorflow.python.distribute.cluster_resolver.tpu.tpu_cluster_resolver import TPUClusterResolv…
|
| H A D | __init__.py | 31 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
|
| H A D | BUILD | 72 srcs = ["tpu_cluster_resolver.py"],
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/ |
| H A D | tpu_initialization_test.py | 20 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
28 resolver = tpu_cluster_resolver.TPUClusterResolver('')
|
| H A D | tpu_embedding_base_test.py | 25 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
151 self.resolver = tpu_cluster_resolver.TPUClusterResolver(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | remote_cloud_tpu_test.py | 20 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
62 resolver = tpu_cluster_resolver.TPUClusterResolver(
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | const_op_test.py | 21 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
37 resolver = tpu_cluster_resolver.TPUClusterResolver(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/ |
| H A D | BUILD | 18 srcs = ["tpu_cluster_resolver.py"],
|
| H A D | tpu_cluster_resolver_test.py | 25 from tensorflow.python.distribute.cluster_resolver.tpu import tpu_cluster_resolver as resolver
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt | 3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.tpu.tpu_cluster_resolver.TPUC…
|
| H A D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 21 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'device_assignment\'], varargs=None, keywords…
|
| H A D | tensorflow.distribute.-t-p-u-strategy.pbtxt | 21 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'experimental_device_assignment\', \'experime…
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt | 3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.tpu.tpu_cluster_resolver.TPUC…
|
| H A D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 25 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'steps_per_run\', \'device_assignment\'], var…
|