| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | collective_all_reduce_strategy.py | 169 cluster_resolver=None, argument
208 def cluster_resolver(self): member in CollectiveAllReduceStrategy
240 cluster_resolver=None): argument
281 cluster_resolver=None): argument
323 def __init__(self, container_strategy, cluster_resolver, argument
352 def _initialize_strategy(self, cluster_resolver, devices): argument
360 def _initialize_local_devices(self, cluster_resolver, worker_device): argument
384 def _initialize_local(self, cluster_resolver, devices=None): argument
451 def _initialize_multi_worker(self, cluster_resolver): argument
|
| H A D | parameter_server_strategy.py | 102 def __init__(self, cluster_resolver=None): argument
165 cluster_resolver=None, argument
179 cluster_resolver=None, argument
188 def _initialize_multi_worker(self, cluster_resolver): argument
284 cluster_resolver=None): argument
|
| H A D | parameter_server_strategy_v2.py | 429 def __init__(self, cluster_resolver, variable_partitioner=None): argument
529 def _verify_args_and_config(self, cluster_resolver): argument
553 def __init__(self, container_strategy, cluster_resolver, argument
|
| H A D | multi_worker_test_base.py | 190 cluster_resolver, argument
324 def cluster_resolver(self): member in MultiProcessCluster
|
| H A D | tpu_strategy.py | 431 def cluster_resolver(self): member in TPUStrategyV2
724 def cluster_resolver(self): member in TPUStrategy
|
| H A D | distribute_lib.py | 1616 def cluster_resolver(self): member in StrategyBase
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_strategy_util.py | 41 def initialize_tpu_system(cluster_resolver=None): argument
162 def shutdown_tpu_system(cluster_resolver=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | collective_ops_multi_worker_test.py | 38 def enable_collective_ops(cluster_resolver): argument
51 def enable_collective_ops_with_barrier(cluster_resolver): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/failure_handling/ |
| H A D | failure_handling.py | 392 cluster_resolver, argument
533 checkpoint_dir, cluster_resolver): argument
|
| H A D | failure_handler_test.py | 63 def _make_checkpoint_manager(checkpoint, checkpoint_dir, cluster_resolver): argument
|
| H A D | gce_failure_handler_test.py | 63 def _make_checkpoint_manager(checkpoint, checkpoint_dir, cluster_resolver): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/ |
| H A D | utils.py | 23 def start_server(cluster_resolver, protocol): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/profiler/ |
| H A D | capture_tpu_profile.py | 82 def get_workers_list(cluster_resolver): argument
|