| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 109 delete worker_env_.rendezvous_mgr; in ~GrpcServer()
116 if (worker_env_.session_mgr != nullptr) { in ~GrpcServer()
117 delete worker_env_.session_mgr; // Deletes graph_mgr's. in ~GrpcServer()
173 worker_env_.env = env_; in Init()
193 worker_env_.device_mgr = new StaticDeviceMgr(std::move(devices)); in Init()
194 owned_device_manager_.reset(worker_env_.device_mgr); in Init()
196 worker_env_.device_mgr = opts.local_device_mgr; in Init()
199 worker_env_.local_devices = worker_env_.device_mgr->ListDevices(); in Init()
200 master_env_.local_devices = worker_env_.device_mgr->ListDevices(); in Init()
201 worker_env_.rendezvous_mgr = opts.rendezvous_mgr_func == nullptr in Init()
[all …]
|
| D | grpc_worker_cache.cc | 41 worker_env_(worker_env), in GrpcWorkerCache()
63 channel, worker_env_->GetCompletionQueue(index), in GetOrCreateWorker()
64 worker_env_->GetThreadPool(), &logger_, target); in GetOrCreateWorker()
101 worker_env_->CompletionQueueSize())) in AssignWorkerToThread()
111 GrpcWorkerEnv* worker_env_; // Not owned member in tensorflow::__anon054a07220111::GrpcWorkerCache
|
| D | grpc_server_lib.h | 104 WorkerEnv* worker_env() { return &worker_env_; } in worker_env()
177 WorkerEnv worker_env_; variable
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.cc | 36 : worker_env_(worker_env), in SessionMgr()
119 CHECK(!worker_env_->local_devices.empty()) in CreateSession()
135 for (Device* d : worker_env_->local_devices) { in CreateSession()
143 AsRemoteDevices(worker_env_->env, cluster_device_attributes, cb, in CreateSession()
152 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, device_mgr.get()); in CreateSession()
159 AsRemoteDevices(worker_env_->env, cluster_device_attributes, nullptr, in CreateSession()
170 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, worker_env_->device_mgr); in CreateSession()
174 worker_env_->device_mgr, std::move(graph_mgr), in CreateSession()
245 AsRemoteDevices(worker_env_->env, added_cluster_device_attrs_pb, nullptr, in UpdateSession()
|
| D | graph_mgr.cc | 60 : worker_env_(worker_env), device_mgr_(device_mgr), table_(5) { in GraphMgr()
140 device_mgr_, worker_env_->env, /*config=*/&config_proto, in InitItem()
142 graph_options.optimizer_options(), worker_env_->compute_pool, cluster_flr, in InitItem()
147 auto* remote_r = this->worker_env_->rendezvous_mgr->Find(step_id); in InitItem()
153 this->worker_env_->rendezvous_mgr->Cleanup(step_id); in InitItem()
273 optimizer.Optimize(lib, worker_env_->env, params.device, &subgraph, in InitItem()
353 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); in SendInputs()
372 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); in RecvOutputs()
391 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); in RecvOutputsAsync()
458 RemoteRendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); in ExecuteAsync()
[all …]
|
| D | session_mgr.h | 94 WorkerEnv* const worker_env_; // Not owned.
|
| D | graph_mgr.h | 147 const WorkerEnv* worker_env_; // Not owned. variable
|
| D | base_rendezvous_mgr.h | 100 const WorkerEnv* const worker_env_; variable
|
| D | base_rendezvous_mgr.cc | 47 : worker_env_(worker_env) {} in BaseRendezvousMgr()
64 auto rr = Create(step_id, worker_env_); in FindOrCreate()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 144 : rendezvous_mgr_(&worker_env_), in EagerServiceImplTest()
146 &worker_env_, "/job:localhost/replica:0/task:0/device:CPU:0", in EagerServiceImplTest()
153 worker_env_.env = Env::Default();
155 worker_env_.rendezvous_mgr = &rendezvous_mgr_;
156 worker_env_.session_mgr = session_mgr_.get();
160 worker_env_.local_devices = device_mgr_->ListDevices();
161 worker_env_.device_mgr = device_mgr_.get();
165 WorkerEnv worker_env_; member in tensorflow::eager::__anon09d935ad0111::EagerServiceImplTest
390 TestEagerServiceImpl eager_service_impl(&worker_env_); in TEST_F()
474 TestEagerServiceImpl eager_service_impl(&worker_env_); in TestFunction()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 506 if (worker_env_ != nullptr && worker_env_->rendezvous_mgr != nullptr) { in CreateRendezvous()
507 auto* remote_r = worker_env_->rendezvous_mgr->Find(step_id); in CreateRendezvous()
682 WorkerEnv* worker_env_ = nullptr; variable
|
| D | context.cc | 1336 worker_env_ = worker_env; in SetMasterContextState()
|