Searched defs:cluster_resolver (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_strategy_util.py | 40 def initialize_tpu_system(cluster_resolver=None): argument
131 def shutdown_tpu_system(cluster_resolver=None): argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy.py | 85 cluster_resolver=None): argument
145 cluster_resolver=None): argument
167 cluster_resolver): argument
178 def _initialize_strategy(self, cluster_resolver): argument
184 def _initialize_local(self, cluster_resolver, devices=None): argument
244 def _initialize_multi_worker(self, cluster_resolver): argument
|
| D | parameter_server_strategy.py | 103 def __init__(self, cluster_resolver=None): argument
129 def __init__(self, cluster_resolver=None): argument
148 cluster_resolver=None, argument
162 cluster_resolver=None, argument
171 def _initialize_multi_worker(self, cluster_resolver): argument
268 cluster_resolver=None): argument
|
| /external/tensorflow/tensorflow/python/tpu/profiler/ |
| D | capture_tpu_profile.py | 84 def get_workers_list(cluster_resolver): argument
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | multi_worker_test.py | 54 def __init__(self, cluster_resolver=None): argument
|