Searched refs:id_in_cluster (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_util_test.py | 161 multi_worker_util.id_in_cluster(cluster_spec, "chief", 0), 0)
170 multi_worker_util.id_in_cluster(cluster_spec, "worker", 1), 2)
177 multi_worker_util.id_in_cluster(cluster_spec, "worker", 1), 1)
186 multi_worker_util.id_in_cluster(cluster_spec, "evaluator", 0), 0)
192 multi_worker_util.id_in_cluster(cluster_spec, "ps", 0)
200 multi_worker_util.id_in_cluster(cluster_spec, "chief", 0)
|
| D | multi_worker_util.py | 192 def id_in_cluster(cluster_spec, task_type, task_id): function
|
| D | parameter_server_strategy.py | 333 input_pipeline_id = multi_worker_util.id_in_cluster(
354 input_pipeline_id = multi_worker_util.id_in_cluster(
|
| D | collective_all_reduce_strategy.py | 400 input_pipeline_id = multi_worker_util.id_in_cluster(
|