Searched refs:collective_leader (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | multi_client_util.py | 33 collective_leader: str,
78 if not collective_leader.startswith("/job:"):
79 collective_leader = "/job:" + collective_leader
82 collective_leader=collective_leader)
85 service_type="standalone", service_leader=collective_leader)
88 config_proto.experimental.collective_group_leader = collective_leader
|
| D | mesh_util.py | 261 collective_leader=api.full_job_name(task_id=0),
|
| D | tpu_util.py | 180 collective_leader=api.full_job_name(task_id=0),
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_util_test.py | 208 multi_worker_util.collective_leader(cluster_spec, "worker", 0),
217 multi_worker_util.collective_leader(cluster_spec, "worker", 1),
228 multi_worker_util.collective_leader(cluster_spec, "evaluator", 0), "")
233 multi_worker_util.collective_leader(cluster_spec, None, 0), "")
|
| D | multi_worker_test_base.py | 192 collective_leader=None): argument
212 if collective_leader:
213 server_config.experimental.collective_group_leader = collective_leader
219 cluster_spec, task_type, task_id, rpc_layer, collective_leader)
334 collective_leader=None): argument
347 collective_leader=collective_leader)
|
| D | multi_worker_util.py | 148 def collective_leader(cluster_spec, task_type, task_id): function
|
| D | collective_all_reduce_strategy.py | 480 collective_leader=multi_worker_util.collective_leader(
774 multi_worker_util.collective_leader(self._cluster_spec, self._task_type,
|
| D | cross_device_ops_test.py | 95 collective_leader="'/job:worker/replica:0/task:0'")
|
| /external/tensorflow/tensorflow/core/distributed_runtime/coordination/ |
| D | coordination_service_agent.cc | 161 const std::string& collective_leader = server_def.default_session_config() in Initialize() local
164 if (!collective_leader.empty()) { in Initialize()
165 configs.set_service_leader(collective_leader); in Initialize()
167 << collective_leader; in Initialize()
|
| D | coordination_service.cc | 79 const std::string& collective_leader = in is_multi_client_leader() local
84 } else if (!collective_leader.empty()) { in is_multi_client_leader()
86 << collective_leader; in is_multi_client_leader()
87 DeviceNameUtils::ParseFullName(collective_leader, &leader_pn); in is_multi_client_leader()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_multi_worker_test.py | 40 collective_leader="/job:worker/replica:0/task:0")
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 830 collective_leader="", argument
854 if (self._collective_leader != collective_leader or
866 self._collective_leader = collective_leader
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 839 collective_leader='/job:worker/replica:0/task:0',
|