Searched refs:worker_session_ (Results 1 – 7 of 7) sorted by relevance
124 worker_session_("rpc_session", "/job:mnist/replica:1/task:2", in RpcRendezvousMgrTest()135 WorkerSession worker_session_; member in tensorflow::RpcRendezvousMgrTest146 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()175 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()189 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()212 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()237 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()268 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()297 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()316 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()
187 std::shared_ptr<WorkerSession> worker_session_; variable
176 worker_session_->worker_cache()->ReleaseWorker(function_data.target, in ~ClusterFunctionLibraryRuntime()190 worker_session_->GetSharedWorkerCache(); in Instantiate()195 worker_session_->worker_cache()->ListWorkers(&workers); in Instantiate()225 req->set_session_handle(worker_session_->session_name()); in Instantiate()276 req->set_session_handle(worker_session_->session_name()); in Run()
34 : worker_session_(worker_session), in ClusterFunctionLibraryRuntime()70 WorkerSession* const worker_session_ = nullptr; // not owned. variable
45 worker_session_.reset(new WorkerSession( in ClusterFunctionLibraryRuntimeTest()50 cluster_flr_.reset(new ClusterFunctionLibraryRuntime(worker_session_.get(), in ClusterFunctionLibraryRuntimeTest()113 std::unique_ptr<WorkerSession> worker_session_; member in tensorflow::ClusterFunctionLibraryRuntimeTest
508 remote_r->Initialize(worker_session_.get()).IgnoreError(); in CreateRendezvous()683 std::shared_ptr<WorkerSession> worker_session_; variable
1337 worker_session_ = std::move(worker_session); in SetMasterContextState()