Searched refs:worker_session_ (Results 1 – 7 of 7) sorted by relevance
76 worker_session_("rpc_session", "/job:mnist/replica:1/task:2", in RpcRendezvousMgrTest()87 WorkerSession worker_session_; member in tensorflow::RpcRendezvousMgrTest98 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()127 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()141 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()164 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()189 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()203 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()236 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()
170 std::shared_ptr<WorkerSession> worker_session_; variable
176 worker_session_->worker_cache()->ReleaseWorker(function_data.target, in ~ClusterFunctionLibraryRuntime()190 worker_session_->worker_cache()->GetOrCreateWorker(target); in Instantiate()194 worker_session_->worker_cache()->ListWorkers(&workers); in Instantiate()224 req->set_session_handle(worker_session_->session_name()); in Instantiate()274 req->set_session_handle(worker_session_->session_name()); in Run()
32 : worker_session_(worker_session), in ClusterFunctionLibraryRuntime()63 WorkerSession* const worker_session_ = nullptr; // not owned. variable
43 worker_session_.reset(new WorkerSession( in ClusterFunctionLibraryRuntimeTest()48 cluster_flr_.reset(new ClusterFunctionLibraryRuntime(worker_session_.get(), in ClusterFunctionLibraryRuntimeTest()111 std::unique_ptr<WorkerSession> worker_session_; member in tensorflow::ClusterFunctionLibraryRuntimeTest
236 remote_r->Initialize(worker_session_.get()).IgnoreError(); in CreateRendezvous()571 std::shared_ptr<WorkerSession> worker_session_; variable
1050 worker_session_ = std::move(worker_session); in SetMasterContextState()