Searched refs:session_mgr (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker.cc | 48 Status s = env_->session_mgr->CreateSession(request->session_handle(), in CreateWorkerSessionAsync()
58 Status s = env_->session_mgr->DeleteSession(request->session_handle()); in DeleteWorkerSessionAsync()
68 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in RegisterGraphAsync()
71 session = env_->session_mgr->LegacySession(); in RegisterGraphAsync()
89 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DeregisterGraphAsync()
92 session = env_->session_mgr->LegacySession(); in DeregisterGraphAsync()
162 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DoRunGraph()
165 session = env_->session_mgr->LegacySession(); in DoRunGraph()
273 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DoPartialRunGraph()
276 session = env_->session_mgr->LegacySession(); in DoPartialRunGraph()
[all …]
|
| D | worker_env.h | 43 SessionMgr* session_mgr = nullptr; member
|
| D | BUILD | 125 name = "session_mgr",
126 srcs = ["session_mgr.cc"],
127 hdrs = ["session_mgr.h"],
145 ":session_mgr",
201 ":session_mgr",
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | BUILD | 58 "//tensorflow/core/distributed_runtime:session_mgr",
116 "//tensorflow/core/distributed_runtime:session_mgr",
142 "//tensorflow/core/distributed_runtime:session_mgr",
|
| D | rdma_mgr.cc | 42 local_worker_ = worker_env_->session_mgr->LegacySession()->worker_name; in RdmaMgr()
44 worker_env_->session_mgr->LegacySession()->worker_cache->ListWorkers( in RdmaMgr()
|
| D | grpc_verbs_service.cc | 144 worker_env_->session_mgr->LegacySession()->worker_name); in GetRemoteAddressSync()
|
| D | rdma.cc | 1031 channel_->adapter_->worker_env_->session_mgr->LegacySession() in PrepareRecvTensor()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 627 auto session_mgr = env->session_mgr; in LoggingAsync() local
628 if (session_mgr) { in LoggingAsync()
630 session_mgr->SetLogging(true); in LoggingAsync()
637 session_mgr->SetLogging(false); in LoggingAsync()
640 session_mgr->RetrieveLogs(step_id, response); in LoggingAsync()
643 session_mgr->ClearLogs(); in LoggingAsync()
|
| D | grpc_server_lib.cc | 100 if (worker_env_.session_mgr != nullptr) { in ~GrpcServer()
101 delete worker_env_.session_mgr; // Deletes graph_mgr's. in ~GrpcServer()
242 worker_env_.session_mgr = new SessionMgr( in Init()
|
| D | BUILD | 303 "//tensorflow/core/distributed_runtime:session_mgr",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | BUILD | 66 "//tensorflow/core/distributed_runtime:session_mgr",
91 "//tensorflow/core/distributed_runtime:session_mgr",
|
| D | eager_service_impl.cc | 110 TF_RETURN_IF_ERROR(env_->session_mgr->CreateSession( in CreateContext()
114 TF_RETURN_IF_ERROR(env_->session_mgr->WorkerSessionForSession( in CreateContext()
|
| D | eager_service_impl_test.cc | 69 worker_env_.session_mgr = session_mgr_.get();
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | BUILD | 76 "//tensorflow/core/distributed_runtime:session_mgr",
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 216 TF_RETURN_IF_ERROR(grpc_server->worker_env()->session_mgr->CreateSession( in UpdateTFE_ContextWithServerDef()
221 grpc_server->worker_env()->session_mgr->WorkerSessionForSession( in UpdateTFE_ContextWithServerDef()
|