| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device.cc | 75 const string& worker_name, NewRemoteDevicesDone done) { in NewRemoteDevices() argument
76 WorkerInterface* wi = worker_cache->GetOrCreateWorker(worker_name); in NewRemoteDevices()
79 done(errors::NotFound("Device ", worker_name, " is not found."), &empty); in NewRemoteDevices()
87 auto cb = [env, worker_cache, worker_name, done, wi, in NewRemoteDevices()
92 [&worker_cache, &worker_name, &wi, &done, &remote_devices, &s, call] { in NewRemoteDevices()
93 worker_cache->ReleaseWorker(worker_name, wi); in NewRemoteDevices()
99 if (!DeviceNameUtils::ParseFullName(worker_name, &worker_name_parsed) || in NewRemoteDevices()
103 worker_name); in NewRemoteDevices()
|
| D | device_resolver_distributed_test.cc | 157 void DefineWorker(const string& worker_name, const string& device_type, in DefineWorker() argument
163 strings::StrCat(worker_name, "/device:", device_type, ":", i), i, in DefineWorker()
168 new TestableDeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
169 resolvers_[worker_name] = dev_res; in DefineWorker()
171 std::vector<string>* dv = &dev_by_task_[worker_name]; in DefineWorker()
176 FakeWorker* fw = new FakeWorker(worker_name, dev_mgr, dev_res); in DefineWorker()
178 wc_.AddWorker(worker_name, fw); in DefineWorker()
181 void RestartWorker(const string& worker_name, const string& device_type, in RestartWorker() argument
190 delete resolvers_[worker_name]; in RestartWorker()
191 DefineWorker(worker_name, device_type, num_devices, in RestartWorker()
[all …]
|
| D | base_rendezvous_mgr.cc | 146 static bool IsLocalDevice(const StringPiece worker_name, in IsLocalDevice() argument
148 return absl::StartsWith(device_name, worker_name); in IsLocalDevice()
157 if (session_->worker_name() == session->worker_name()) { in Initialize()
163 session_->worker_name(), " -> ", session->worker_name()); in Initialize()
198 if (!IsLocalDevice(sess->worker_name(), parsed.src_device)) { in Send()
201 sess->worker_name()); in Send()
221 if (is_src && !IsLocalDevice(sess->worker_name(), parsed.src_device)) { in ValidateDevices()
224 sess->worker_name()); in ValidateDevices()
226 if (!is_src && !IsLocalDevice(sess->worker_name(), parsed.dst_device)) { in ValidateDevices()
229 sess->worker_name()); in ValidateDevices()
|
| D | worker_session.h | 47 const string& worker_name() const { return worker_name_; } in worker_name() function
56 WorkerSession(const string& session_name, const string& worker_name,
63 const string& session_name, const string& worker_name,
79 WorkerSession(const string& session_name, const string& worker_name,
|
| D | collective_rma_distributed_test.cc | 217 void DefineWorker(const string& worker_name, const string& device_type, in DefineWorker() argument
223 strings::StrCat(worker_name, "/device:", device_type, ":", i))); in DefineWorker()
227 std::vector<string>* dv = &dev_by_task_[worker_name]; in DefineWorker()
233 new DeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
234 dev_resolvers_[worker_name] = dev_res; in DefineWorker()
235 FakeWorker* fw = new FakeWorker(worker_name, dev_mgr, dev_res); in DefineWorker()
237 wc_.AddWorker(worker_name, fw); in DefineWorker()
240 void RestartWorker(const string& worker_name, const string& device_type, in RestartWorker() argument
242 auto it = dev_resolvers_.find(worker_name); in RestartWorker()
247 DefineWorker(worker_name, device_type, num_devices); in RestartWorker()
|
| D | worker_session.cc | 108 const string& session_name, const string& worker_name, in WorkerSession() argument
113 worker_name_(worker_name), in WorkerSession()
146 const string& session_name, const string& worker_name, in CreateWithBorrowedDeviceMgr() argument
151 session_name, worker_name, std::move(worker_cache), borrowed_device_mgr, in CreateWithBorrowedDeviceMgr()
156 const string& session_name, const string& worker_name, in WorkerSession() argument
161 worker_name_(worker_name), in WorkerSession()
|
| D | collective_param_resolver_distributed_test.cc | 160 void DefineWorker(const ConfigProto& config, const string& worker_name, in DefineWorker() argument
166 strings::StrCat(worker_name, "/device:", device_type, ":", i))); in DefineWorker()
170 std::vector<string>* dv = &dev_by_task_[worker_name]; in DefineWorker()
175 new DeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
176 dev_resolvers_[worker_name] = dev_res; in DefineWorker()
179 worker_name); in DefineWorker()
180 cp_resolvers_[worker_name] = cp_res; in DefineWorker()
181 FakeWorker* fw = new FakeWorker(worker_name, dev_mgr, cp_res); in DefineWorker()
183 wc_.AddWorker(worker_name, fw); in DefineWorker()
|
| D | session_mgr.cc | 71 string worker_name; in CreateSession() local
74 worker_name = legacy_session_->worker_name(); in CreateSession()
77 worker_name = WorkerNameFromServerDef(server_def); in CreateSession()
102 worker_name, d, false, isolate_session_state)); in CreateSession()
119 new WorkerSession(session, worker_name, in CreateSession()
137 session, worker_name, in CreateSession()
229 legacy_session_->worker_name(), in WorkerSessionForSessionLocked()
|
| D | session_mgr_test.cc | 105 EXPECT_EQ("/job:worker/replica:0/task:3", session->worker_name()); in TEST_F()
116 EXPECT_EQ("/job:mnist/replica:0/task:0", session->worker_name()); in TEST_F()
177 string worker_name = SessionMgr::WorkerNameFromServerDef(server_def); in TEST_F() local
178 EXPECT_EQ("/job:worker/replica:0/task:3", worker_name); in TEST_F()
|
| D | remote_device.h | 61 const string& worker_name, NewRemoteDevicesDone done);
|
| D | master.cc | 631 const string& worker_name = worker_names[i]; in CleanupWorkers() local
632 auto worker = env_->worker_cache->GetOrCreateWorker(worker_name); in CleanupWorkers()
635 &req, &resp[i], [this, &n, worker_name, worker, c](Status s) { in CleanupWorkers()
637 env_->worker_cache->ReleaseWorker(worker_name, worker); in CleanupWorkers()
|
| D | master_session.cc | 509 const string* worker_name; member
526 strings::StrCat("From ", *call->worker_name, ":\n", in WhenDone()
531 Status(s.code(), strings::StrCat("From ", *call->worker_name, ":\n", in WhenDone()
567 << *call.worker_name; in Wait()
673 c->worker_name = &part.name; in RunPartitionsHelper()
|
| D | worker.cc | 445 env_->session_mgr->LegacySession()->worker_name(), in PrepareRecvTensor()
|
| /external/libchrome/base/task_scheduler/ |
| D | scheduler_single_thread_task_runner_manager.cc | 490 std::string worker_name; in CreateTaskRunnerWithTraitsImpl() local
492 worker_name += "Shared"; in CreateTaskRunnerWithTraitsImpl()
493 worker_name += environment_params.name_suffix; in CreateTaskRunnerWithTraitsImpl()
495 worker_name, thread_mode, in CreateTaskRunnerWithTraitsImpl()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 160 << port::Hostname() << " " << worker_session->worker_name(); in CreateContext()
175 worker_session->worker_name()), in CreateContext()
274 worker_session->worker_name()), in UpdateContext()
276 VLOG(1) << "On existing server " << worker_session->worker_name() in UpdateContext()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 462 string worker_name = in UpdateTFE_ContextWithServerDef() local
480 ListRemoteWorkers(grpc_server, worker_name, &remote_workers)); in UpdateTFE_ContextWithServerDef()
482 LOG_AND_RETURN_IF_ERROR(ListRemoteWorkers(context->GetServer(), worker_name, in UpdateTFE_ContextWithServerDef()
489 ListRemoteWorkers(grpc_server, worker_name, &remote_workers)); in UpdateTFE_ContextWithServerDef()
824 const char* worker_name, in TFE_ContextCheckAlive() argument
845 status->status = remote_eager_workers->GetClient(worker_name, &eager_client); in TFE_ContextCheckAlive()
874 LOG(INFO) << "Remote worker " << worker_name in TFE_ContextCheckAlive()
|
| D | c_api_experimental.h | 434 const char* worker_name,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 604 def check_alive(self, worker_name): argument
619 return pywrap_tfe.TFE_ContextCheckAlive(self._context_handle, worker_name)
2097 def check_alive(worker_name): argument
2098 return context().check_alive(worker_name)
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 462 m.def("TFE_ContextCheckAlive", [](py::handle& ctx, const char* worker_name) { in PYBIND11_MODULE() argument
466 worker_name, status.get()); in PYBIND11_MODULE()
|