Searched refs:WorkerSessionForSession (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr_test.cc | 73 TF_EXPECT_OK(mgr_.WorkerSessionForSession(session_handle, &session)); in TEST_F()
97 TF_EXPECT_OK(mgr_.WorkerSessionForSession(session_handle, &session)); in TEST_F()
114 TF_EXPECT_OK(mgr_.WorkerSessionForSession(session_handle, &session)); in TEST_F()
127 TF_EXPECT_OK(mgr_.WorkerSessionForSession("handle_1", &session_1)); in TEST_F()
133 TF_EXPECT_OK(mgr_.WorkerSessionForSession("handle_2", &session_2)); in TEST_F()
139 TF_EXPECT_OK(mgr_.WorkerSessionForSession("handle_3", &session_3)); in TEST_F()
145 TF_EXPECT_OK(mgr_.WorkerSessionForSession("handle_4", &session_4)); in TEST_F()
180 TF_EXPECT_OK(mgr_.WorkerSessionForSession(sess_handle1, &session)); in TEST_F()
183 TF_EXPECT_OK(mgr_.WorkerSessionForSession(sess_handle2, &session)); in TEST_F()
193 EXPECT_NE(mgr_.WorkerSessionForSession(sess_handle1, &session), in TEST_F()
[all …]
|
| D | worker.cc | 75 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in RegisterGraphAsync()
96 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DeregisterGraphAsync()
175 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DoRunGraph()
277 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DoPartialRunGraph()
|
| D | session_mgr.h | 79 Status WorkerSessionForSession(const string& session_handle,
|
| D | session_mgr.cc | 281 Status SessionMgr::WorkerSessionForSession( in WorkerSessionForSession() function in tensorflow::SessionMgr
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 255 TF_RETURN_IF_ERROR(env_->session_mgr->WorkerSessionForSession( in CreateContext()
370 TF_RETURN_IF_ERROR(env_->session_mgr->WorkerSessionForSession( in UpdateContext()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_distributed_manager.cc | 615 grpc_server->worker_env()->session_mgr->WorkerSessionForSession( in UpdateContextWithServerDef()
|