Home
last modified time | relevance | path

Searched refs:id_in_cluster (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_util_test.py161 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)
Dinput_lib_test.py622 id_in_cluster = multi_worker_util.id_in_cluster(cr.cluster_spec(),
631 return dataset.shard(worker_count, id_in_cluster).batch(1)
639 if id_in_cluster == 0:
645 if id_in_cluster == 0:
683 id_in_cluster = multi_worker_util.id_in_cluster(cr.cluster_spec(),
692 return dataset.shard(worker_count, id_in_cluster).batch(1)
701 if id_in_cluster == 0:
706 if id_in_cluster == 0:
926 id_in_cluster = multi_worker_util.id_in_cluster(cr.cluster_spec(),
937 if id_in_cluster == 0:
[all …]
Dmulti_worker_util.py207 def id_in_cluster(cluster_spec, task_type, task_id): function
Dparameter_server_strategy.py361 input_pipeline_id = multi_worker_util.id_in_cluster(
382 input_pipeline_id = multi_worker_util.id_in_cluster(
Dcollective_all_reduce_strategy.py419 self._id_in_cluster = multi_worker_util.id_in_cluster(