Searched defs:WorkerGroup (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/elastic/agent/server/ |
| H A D | api.py | 244 class WorkerGroup: class
451 def _start_workers(self, worker_group: WorkerGroup) -> Dict[int, Any]:
461 self, worker_group: WorkerGroup, is_restart: bool = False
472 def _monitor_workers(self, worker_group: WorkerGroup) -> RunResult:
491 def _rendezvous(self, worker_group: WorkerGroup) -> None:
650 def _initialize_workers(self, worker_group: WorkerGroup) -> None:
681 def _restart_workers(self, worker_group: WorkerGroup) -> None:
|
| H A D | local_elastic_agent.py | 277 self, worker_group: WorkerGroup, is_restart: bool = False
284 def _start_workers(self, worker_group: WorkerGroup) -> Dict[int, Any]:
372 def _monitor_workers(self, worker_group: WorkerGroup) -> RunResult:
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | master_session.cc | 1294 struct WorkerGroup { in CreateWorkerSessions() struct
1296 const string* name; in CreateWorkerSessions()
1299 WorkerInterface* worker = nullptr; in CreateWorkerSessions()
1302 CreateWorkerSessionRequest request; in CreateWorkerSessions()
1303 CreateWorkerSessionResponse response; in CreateWorkerSessions()
1304 Status status = OkStatus(); in CreateWorkerSessions()
1427 struct WorkerGroup { in DeleteWorkerSessions() struct
1429 const string* name; in DeleteWorkerSessions()
1432 WorkerInterface* worker = nullptr; in DeleteWorkerSessions()
1434 CallOptions call_opts; in DeleteWorkerSessions()
[all …]
|
| /aosp_15_r20/external/pytorch/test/distributed/elastic/agent/server/test/ |
| H A D | api_test.py | 129 self, worker_group: WorkerGroup, is_restart: bool = False
136 def _start_workers(self, worker_group: WorkerGroup) -> Dict[int, Any]:
144 def _monitor_workers(self, worker_group: WorkerGroup) -> RunResult:
|