| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device.cc | 58 const string& worker_name, NewRemoteDevicesDone done) { in NewRemoteDevices() argument
59 WorkerInterface* wi = worker_cache->CreateWorker(worker_name); in NewRemoteDevices()
62 done(errors::NotFound("Device ", worker_name, " is not found."), &empty); in NewRemoteDevices()
70 auto cb = [env, worker_cache, worker_name, done, wi, in NewRemoteDevices()
75 [&worker_cache, &worker_name, &wi, &done, &remote_devices, &s, call] { in NewRemoteDevices()
76 worker_cache->ReleaseWorker(worker_name, wi); in NewRemoteDevices()
82 if (!DeviceNameUtils::ParseFullName(worker_name, &worker_name_parsed) || in NewRemoteDevices()
86 worker_name); in NewRemoteDevices()
|
| D | worker_session.cc | 96 const string& worker_name, in WorkerSession() argument
101 worker_name(worker_name), in WorkerSession()
111 const string& session_name, const string& worker_name, in CreateWithBorrowedDeviceMgr() argument
115 new WorkerSession(session_name, worker_name, std::move(worker_cache), in CreateWithBorrowedDeviceMgr()
120 const string& worker_name, in WorkerSession() argument
125 worker_name(worker_name), in WorkerSession()
|
| D | session_mgr.cc | 61 string worker_name; in CreateSession() local
64 worker_name = legacy_session_->worker_name; in CreateSession()
67 worker_name = WorkerNameFromServerDef(server_def); in CreateSession()
84 worker_name, d, false, isolate_session_state)); in CreateSession()
90 new WorkerSession(session, worker_name, in CreateSession()
99 session, worker_name, in CreateSession()
126 legacy_session_->worker_name, in WorkerSessionForSessionLocked()
|
| D | base_rendezvous_mgr.cc | 147 static bool IsLocalDevice(const StringPiece worker_name, in IsLocalDevice() argument
149 return str_util::StartsWith(device_name, worker_name); in IsLocalDevice()
158 if (session_->worker_name == session->worker_name) { in Initialize()
164 session_->worker_name, " -> ", session->worker_name); in Initialize()
195 if (!IsLocalDevice(session_->worker_name, parsed.src_device)) { in Send()
198 session_->worker_name); in Send()
218 if (is_src && !IsLocalDevice(sess->worker_name, parsed.src_device)) { in ValidateDevices()
220 parsed.FullKey(), " @ ", sess->worker_name); in ValidateDevices()
222 if (!is_src && !IsLocalDevice(sess->worker_name, parsed.dst_device)) { in ValidateDevices()
224 parsed.FullKey(), " @ ", sess->worker_name); in ValidateDevices()
|
| D | worker_session.h | 38 const string worker_name; member
60 WorkerSession(const string& session_name, const string& worker_name,
66 const string& session_name, const string& worker_name,
73 WorkerSession(const string& session_name, const string& worker_name,
|
| D | device_resolver_distributed_test.cc | 152 void DefineWorker(const string& worker_name, const string& device_type, in DefineWorker() argument
158 strings::StrCat(worker_name, "/device:", device_type, ":", i), i)); in DefineWorker()
162 new TestableDeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
163 resolvers_[worker_name] = dev_res; in DefineWorker()
165 std::vector<string>* dv = &dev_by_task_[worker_name]; in DefineWorker()
169 FakeWorker* fw = new FakeWorker(worker_name, dev_mgr, dev_res); in DefineWorker()
171 wc_.AddWorker(worker_name, fw); in DefineWorker()
|
| D | collective_param_resolver_distributed_test.cc | 157 void DefineWorker(const ConfigProto& config, const string& worker_name, in DefineWorker() argument
163 strings::StrCat(worker_name, "/device:", device_type, ":", i))); in DefineWorker()
167 std::vector<string>* dv = &dev_by_task_[worker_name]; in DefineWorker()
172 new DeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
173 dev_resolvers_[worker_name] = dev_res; in DefineWorker()
176 worker_name); in DefineWorker()
177 cp_resolvers_[worker_name] = cp_res; in DefineWorker()
178 FakeWorker* fw = new FakeWorker(worker_name, dev_mgr, cp_res); in DefineWorker()
180 wc_.AddWorker(worker_name, fw); in DefineWorker()
|
| D | collective_rma_distributed_test.cc | 212 void DefineWorker(const ConfigProto& config, const string& worker_name, in DefineWorker() argument
218 strings::StrCat(worker_name, "/device:", device_type, ":", i))); in DefineWorker()
222 std::vector<string>* dv = &dev_by_task_[worker_name]; in DefineWorker()
227 new DeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
228 dev_resolvers_[worker_name] = dev_res; in DefineWorker()
229 FakeWorker* fw = new FakeWorker(worker_name, dev_mgr, dev_res); in DefineWorker()
231 wc_.AddWorker(worker_name, fw); in DefineWorker()
|
| D | session_mgr_test.cc | 92 EXPECT_EQ("/job:worker/replica:0/task:3", session->worker_name); in TEST_F()
103 EXPECT_EQ("/job:mnist/replica:0/task:0", session->worker_name); in TEST_F()
164 string worker_name = SessionMgr::WorkerNameFromServerDef(server_def); in TEST_F() local
165 EXPECT_EQ("/job:worker/replica:0/task:3", worker_name); in TEST_F()
|
| D | master.cc | 631 const string& worker_name = worker_names[i]; in CleanupWorkers() local
632 auto worker = env_->worker_cache->CreateWorker(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 | worker.cc | 443 env_->session_mgr->LegacySession()->worker_name, in PrepareRecvTensor()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_mgr.cc | 42 local_worker_ = worker_env_->session_mgr->LegacySession()->worker_name; in RdmaMgr()
62 string worker_name = p.first; in SetupChannels() local
63 RDMA_LOG(2) << "Connecting to remote node " << worker_name; in SetupChannels()
69 channel_cache_->FindWorkerChannel(worker_name); in SetupChannels()
71 CHECK(client != nullptr) << "No worker known as " << worker_name; in SetupChannels()
96 CHECK(worker_name.compare(resp.host_name()) == 0); in SetupChannels()
121 LOG(ERROR) << "Connecting to " << worker_name << ": Got " in SetupChannels()
130 RDMA_LOG(0) << "Connected to remote node " << worker_name; in SetupChannels()
140 string worker_name = p.first; in ConnectivityCheck() local
143 VLOG(2) << "Ping to " << worker_name; in ConnectivityCheck()
[all …]
|
| D | grpc_verbs_service.cc | 144 worker_env_->session_mgr->LegacySession()->worker_name); in GetRemoteAddressSync()
|
| D | rdma.cc | 1032 ->worker_name, in PrepareRecvTensor()
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_utils.cc | 23 MPIUtils::MPIUtils(const std::string& worker_name) { in MPIUtils() argument
33 CHECK(worker_name.size() < max_worker_name_length) in MPIUtils()
35 snprintf(my_name, max_worker_name_length, worker_name.c_str()); in MPIUtils()
|
| D | mpi_utils.h | 45 explicit MPIUtils(const std::string& worker_name);
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 245 return TPUAssignment(worker_name=self._worker_name, num_cores=num_cores)
256 def __init__(self, worker_name, num_cores): argument
257 self._worker_name = worker_name
261 def worker_name(self): member in TPUAssignment
388 def _clone_optimizer(optimizer, config=None, worker_name=None): argument
404 '%s/device:CPU:0' % ('/job:%s' % worker_name if worker_name else '')):
685 '/job:%s/device:CPU:0' % self._tpu_assignment.worker_name):
840 '/job:%s/device:CPU:0' % self._tpu_assignment.worker_name):
993 worker_name=self._tpu_assignment.worker_name)
1106 '/job:%s/device:CPU:0' % self._tpu_assignment.worker_name):
[all …]
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 172 string worker_name = in UpdateTFE_ContextWithServerDef() local
193 std::remove(remote_workers.begin(), remote_workers.end(), worker_name), in UpdateTFE_ContextWithServerDef()
|