Searched refs:collective_leader (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_util_test.py | 212 multi_worker_util.collective_leader(cluster_spec, "worker", 0),
221 multi_worker_util.collective_leader(cluster_spec, "worker", 1),
232 multi_worker_util.collective_leader(cluster_spec, "evaluator", 0), "")
237 multi_worker_util.collective_leader(cluster_spec, None, 0), "")
|
| D | collective_all_reduce_strategy.py | 271 collective_leader=multi_worker_util.collective_leader(
498 multi_worker_util.collective_leader(self._cluster_spec, self._task_type,
|
| D | multi_worker_util.py | 137 def collective_leader(cluster_spec, task_type, task_id): function
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 647 collective_leader="", argument
671 if (self._collective_leader != collective_leader or
683 self._collective_leader = collective_leader
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 673 collective_leader='/job:worker/replica:0/task:0',
|