Searched refs:collective_leader (Results 1 – 7 of 7) 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 | multi_worker_util.py | 152 def collective_leader(cluster_spec, task_type, task_id): function
|
| D | collective_all_reduce_strategy.py | 436 collective_leader=multi_worker_util.collective_leader(
716 multi_worker_util.collective_leader(self._cluster_spec, self._task_type,
|
| D | cross_device_ops_test.py | 99 collective_leader="'/job:worker/replica:0/task:0'")
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_multi_worker_test.py | 44 collective_leader="/job:worker/replica:0/task:0")
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 705 collective_leader="", argument
729 if (self._collective_leader != collective_leader or
741 self._collective_leader = collective_leader
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 786 collective_leader='/job:worker/replica:0/task:0',
|