Searched refs:remote_workers (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 172 std::vector<string> remote_workers; in CreateContext() local
173 worker_session->worker_cache()->ListWorkers(&remote_workers); in CreateContext()
174 remote_workers.erase(std::remove(remote_workers.begin(), remote_workers.end(), in CreateContext()
176 remote_workers.end()); in CreateContext()
188 remote_workers, request->context_id(), request->context_view_id(), in CreateContext()
271 std::vector<string> remote_workers; in UpdateContext() local
272 worker_session->worker_cache()->ListWorkers(&remote_workers); in UpdateContext()
273 remote_workers.erase(std::remove(remote_workers.begin(), remote_workers.end(), in UpdateContext()
275 remote_workers.end()); in UpdateContext()
279 for (const string& rw : remote_workers) { in UpdateContext()
[all …]
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 150 const std::vector<string>& remote_workers, in GetAllRemoteDevices() argument
154 TF_RETURN_IF_ERROR(AddRemoteDevicesToMgr(remote_workers, worker_cache, in GetAllRemoteDevices()
181 std::vector<string>* remote_workers) { in ListRemoteWorkers() argument
188 grpc_server->master_env()->worker_cache->ListWorkers(remote_workers); in ListRemoteWorkers()
189 remote_workers->erase( in ListRemoteWorkers()
190 std::remove(remote_workers->begin(), remote_workers->end(), local_worker), in ListRemoteWorkers()
191 remote_workers->end()); in ListRemoteWorkers()
269 TFE_Context* ctx, const std::vector<string>& remote_workers, in CreateRemoteContexts() argument
275 int num_remote_workers = remote_workers.size(); in CreateRemoteContexts()
279 const string& remote_worker = remote_workers[i]; in CreateRemoteContexts()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 546 const std::vector<string>& remote_workers) { in RegisterExistingFunctionsOnRemoteWorkers() argument
550 for (int i = 0; i < remote_workers.size(); i++) { in RegisterExistingFunctionsOnRemoteWorkers()
553 remote_eager_workers_->GetClient(remote_workers[i], &eager_client); in RegisterExistingFunctionsOnRemoteWorkers()
|
| D | context.h | 428 const std::vector<string>& remote_workers);
|