Home
last modified time | relevance | path

Searched refs:worker_session (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc254 std::shared_ptr<WorkerSession> worker_session; in CreateContext() local
256 session_name, &worker_session)); in CreateContext()
258 const tensorflow::DeviceMgr* device_mgr = worker_session->device_mgr(); in CreateContext()
261 TF_RETURN_IF_ERROR(r->Initialize(worker_session.get())); in CreateContext()
264 [worker_session, this](const int64 step_id) { in CreateContext()
266 r->Initialize(worker_session.get()).IgnoreError(); in CreateContext()
272 << port::Hostname() << " " << worker_session->worker_name(); in CreateContext()
277 request->async(), device_mgr, false, r, worker_session->cluster_flr()); in CreateContext()
283 worker_session->worker_cache()->ListWorkers(&remote_workers); in CreateContext()
285 worker_session->worker_name()), in CreateContext()
[all …]
Dcluster_function_library_runtime.h108 const uint64 context_id, EagerContext* ctx, WorkerSession* worker_session);
Dcluster_function_library_runtime.cc286 const uint64 context_id, EagerContext* ctx, WorkerSession* worker_session) { in CreateClusterFLR() argument
288 context_id, ctx, worker_session->remote_device_mgr()); in CreateClusterFLR()
DBUILD50 "//tensorflow/core/distributed_runtime:worker_session",
/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr.cc122 std::shared_ptr<WorkerSession> worker_session; in CreateSession() local
153 worker_session.reset( in CreateSession()
171 worker_session = WorkerSession::CreateWithBorrowedDeviceMgr( in CreateSession()
178 sessions_.insert(std::make_pair(session, std::move(worker_session))); in CreateSession()
204 std::shared_ptr<WorkerSession> worker_session = it->second; in UpdateSession() local
217 const DeviceMgr* local_device_mgr = worker_session->device_mgr(); in UpdateSession()
218 DeviceMgr* remote_device_mgr = worker_session->remote_device_mgr(); in UpdateSession()
248 TF_RETURN_IF_ERROR(worker_session->UpdateWorkerCacheAndDevices( in UpdateSession()
Dcluster_function_library_runtime.h31 ClusterFunctionLibraryRuntime(WorkerSession* worker_session, in ClusterFunctionLibraryRuntime() argument
34 : worker_session_(worker_session), in ClusterFunctionLibraryRuntime()
DBUILD89 name = "worker_session",
92 "worker_session.cc",
96 "worker_session.h",
120 ":worker_session",
142 ":worker_session",
216 ":worker_session",
408 ":worker_session",
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext_distributed_manager.cc609 std::shared_ptr<tensorflow::WorkerSession> worker_session; in UpdateContextWithServerDef() local
616 session_name, &worker_session)); in UpdateContextWithServerDef()
619 LOG_AND_RETURN_IF_ERROR(r->Initialize(worker_session.get())); in UpdateContextWithServerDef()
623 worker_session.get()); in UpdateContextWithServerDef()
628 std::move(new_server), grpc_server->worker_env(), worker_session, in UpdateContextWithServerDef()
Dcontext.h367 std::shared_ptr<WorkerSession> worker_session,
669 std::shared_ptr<WorkerSession> worker_session,
Dcontext.cc1202 std::shared_ptr<WorkerSession> worker_session, in InitializeRemoteMaster() argument
1225 std::move(server), worker_env, std::move(worker_session), in InitializeRemoteMaster()
1300 std::shared_ptr<WorkerSession> worker_session, in SetMasterContextState() argument
1337 worker_session_ = std::move(worker_session); in SetMasterContextState()
DBUILD119 "//tensorflow/core/distributed_runtime:worker_session",
217 "//tensorflow/core/distributed_runtime:worker_session",
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
DBUILD225 "//tensorflow/core/distributed_runtime:worker_session",