Searched refs:session_mgr (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker.cc | 54 Status s = env_->session_mgr->CreateSession( in CreateWorkerSessionAsync()
64 Status s = env_->session_mgr->DeleteSession(request->session_handle()); in DeleteWorkerSessionAsync()
74 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in RegisterGraphAsync()
77 session = env_->session_mgr->LegacySession(); in RegisterGraphAsync()
95 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DeregisterGraphAsync()
98 session = env_->session_mgr->LegacySession(); in DeregisterGraphAsync()
174 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DoRunGraph()
177 session = env_->session_mgr->LegacySession(); in DoRunGraph()
274 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DoPartialRunGraph()
277 session = env_->session_mgr->LegacySession(); in DoPartialRunGraph()
[all …]
|
| D | worker_env.h | 43 SessionMgr* session_mgr = nullptr; member
|
| D | BUILD | 133 name = "session_mgr",
134 srcs = ["session_mgr.cc"],
135 hdrs = ["session_mgr.h"],
154 ":session_mgr",
211 ":session_mgr",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 703 auto session_mgr = env->session_mgr; in LoggingAsync() local
704 if (session_mgr) { in LoggingAsync()
706 session_mgr->SetLogging(true); in LoggingAsync()
713 session_mgr->SetLogging(false); in LoggingAsync()
716 session_mgr->RetrieveLogs(step_id, response); in LoggingAsync()
719 session_mgr->ClearLogs(); in LoggingAsync()
|
| D | grpc_server_lib.cc | 119 if (worker_env_.session_mgr != nullptr) { in ~GrpcServer()
120 delete worker_env_.session_mgr; // Deletes graph_mgr's. in ~GrpcServer()
275 worker_env_.session_mgr = new SessionMgr( in Init()
|
| D | BUILD | 314 "//tensorflow/core/distributed_runtime:session_mgr",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 129 TF_RETURN_IF_ERROR(env_->session_mgr->CreateSession( in CreateContext()
135 auto s = env_->session_mgr->DeleteSession(session_name); in CreateContext()
143 TF_RETURN_IF_ERROR(env_->session_mgr->WorkerSessionForSession( in CreateContext()
261 TF_RETURN_IF_ERROR(env_->session_mgr->UpdateSession( in UpdateContext()
266 TF_RETURN_IF_ERROR(env_->session_mgr->WorkerSessionForSession( in UpdateContext()
|
| D | BUILD | 116 "//tensorflow/core/distributed_runtime:session_mgr",
147 "//tensorflow/core/distributed_runtime:session_mgr",
|
| D | eager_service_impl_test.cc | 144 worker_env_.session_mgr = session_mgr_.get();
|
| /external/tensorflow/tensorflow/c/experimental/ |
| D | network_test.cc | 171 TF_EXPECT_OK(grpc_server->worker_env()->session_mgr->CreateSession( in InitializeRendezvous()
175 TF_EXPECT_OK(grpc_server->worker_env()->session_mgr->WorkerSessionForSession( in InitializeRendezvous()
|
| D | BUILD | 116 "//tensorflow/core/distributed_runtime:session_mgr",
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 628 TF_RETURN_IF_ERROR(grpc_server->worker_env()->session_mgr->CreateSession( in UpdateTFE_ContextWithServerDef()
632 grpc_server->worker_env()->session_mgr->WorkerSessionForSession( in UpdateTFE_ContextWithServerDef()
655 grpc_server->worker_env()->session_mgr->UpdateSession( in UpdateTFE_ContextWithServerDef()
659 grpc_server->worker_env()->session_mgr->WorkerSessionForSession( in UpdateTFE_ContextWithServerDef()
|