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()
90 TF_EXPECT_OK(mgr_.WorkerSessionForSession(session_handle, &session)); in TEST_F()
101 TF_EXPECT_OK(mgr_.WorkerSessionForSession(session_handle, &session)); in TEST_F()
114 TF_EXPECT_OK(mgr_.WorkerSessionForSession("handle_1", &session_1)); in TEST_F()
120 TF_EXPECT_OK(mgr_.WorkerSessionForSession("handle_2", &session_2)); in TEST_F()
126 TF_EXPECT_OK(mgr_.WorkerSessionForSession("handle_3", &session_3)); in TEST_F()
132 TF_EXPECT_OK(mgr_.WorkerSessionForSession("handle_4", &session_4)); in TEST_F()
145 TF_EXPECT_OK(mgr_.WorkerSessionForSession(session_handle, &session)); in TEST_F()
154 Status s = mgr_.WorkerSessionForSession(session_handle, &session); in TEST_F()
|
| D | worker.cc | 68 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in RegisterGraphAsync()
89 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DeregisterGraphAsync()
162 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DoRunGraph()
273 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), in DoPartialRunGraph()
|
| D | session_mgr.h | 53 Status WorkerSessionForSession(const string& session_handle,
|
| D | session_mgr.cc | 135 Status SessionMgr::WorkerSessionForSession( in WorkerSessionForSession() function in tensorflow::SessionMgr
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 114 TF_RETURN_IF_ERROR(env_->session_mgr->WorkerSessionForSession( in CreateContext()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 221 grpc_server->worker_env()->session_mgr->WorkerSessionForSession( in UpdateTFE_ContextWithServerDef()
|