| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_strategy_util.py | 41 def initialize_tpu_system(cluster_resolver=None): argument
143 def shutdown_tpu_system(cluster_resolver=None): argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy.py | 168 cluster_resolver=None, argument
207 def cluster_resolver(self): member in CollectiveAllReduceStrategy
240 cluster_resolver=None): argument
281 cluster_resolver=None): argument
316 def __init__(self, container_strategy, cluster_resolver, argument
334 def _initialize_strategy(self, cluster_resolver): argument
340 def _initialize_local(self, cluster_resolver, devices=None): argument
407 def _initialize_multi_worker(self, cluster_resolver): argument
|
| D | parameter_server_strategy_v2.py | 393 def __init__(self, cluster_resolver, variable_partitioner=None): argument
485 def _verify_args_and_config(self, cluster_resolver): argument
514 def __init__(self, container_strategy, cluster_resolver, argument
|
| D | parameter_server_strategy.py | 104 def __init__(self, cluster_resolver=None): argument
167 cluster_resolver=None, argument
181 cluster_resolver=None, argument
190 def _initialize_multi_worker(self, cluster_resolver): argument
286 cluster_resolver=None): argument
|
| D | multi_worker_test_base.py | 222 def __init__(self, cluster_resolver): argument
349 def cluster_resolver(self): member in MultiProcessCluster
|
| D | tpu_strategy.py | 668 def cluster_resolver(self): member in TPUStrategy
|
| D | distribute_lib.py | 1559 def cluster_resolver(self): member in StrategyBase
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | utils.py | 28 def start_server(cluster_resolver, protocol): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_multi_worker_test.py | 42 def enable_collective_ops(cluster_resolver): argument
55 def enable_collective_ops_with_barrier(cluster_resolver): argument
|
| /external/tensorflow/tensorflow/python/tpu/profiler/ |
| D | capture_tpu_profile.py | 85 def get_workers_list(cluster_resolver): argument
|