Searched refs:worker_session_ (Results 1 – 5 of 5) sorted by relevance
71 worker_session_("rpc_session", "/job:mnist/replica:1/task:2", in RpcRendezvousMgrTest()82 WorkerSession worker_session_; member in tensorflow::RpcRendezvousMgrTest93 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()122 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()136 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()148 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()181 TF_ASSERT_OK(rendez->Initialize(&worker_session_)); in TEST_F()
164 std::shared_ptr<WorkerSession> worker_session_; variable
115 worker_session_->worker_cache->ReleaseWorker(function_data.target, in ~ClusterFunctionLibraryRuntime()127 worker_session_->worker_cache->CreateWorker(options.target); in Instantiate()131 worker_session_->worker_cache->ListWorkers(&workers); in Instantiate()147 req.set_session_handle(worker_session_->session_name); in Instantiate()185 req->set_session_handle(worker_session_->session_name); in Run()
32 : worker_session_(worker_session), in ClusterFunctionLibraryRuntime()55 WorkerSession* const worker_session_ = nullptr; // not owned. variable
42 worker_session_.reset(new WorkerSession( in ClusterFunctionLibraryRuntimeTest()48 new ClusterFunctionLibraryRuntime(worker_session_.get(), true)); in ClusterFunctionLibraryRuntimeTest()100 std::unique_ptr<WorkerSession> worker_session_; member in tensorflow::ClusterFunctionLibraryRuntimeTest