diff options
author | 2017-10-02 11:10:44 -0700 | |
---|---|---|
committer | 2017-10-02 11:14:41 -0700 | |
commit | 9bfa43625061ec62bd9623ab014db4851307e92d (patch) | |
tree | cd59c3f044646b3bf9663b8dd8a4250b525e3d27 /tensorflow/core/distributed_runtime/session_mgr.cc | |
parent | c0644791cfc064d5e4652271e51d826aeccad0c2 (diff) |
Allowing for functions to run across processes using RPC's. Currently this only works for processes running on CPU's only.
PiperOrigin-RevId: 170725482
Diffstat (limited to 'tensorflow/core/distributed_runtime/session_mgr.cc')
-rw-r--r-- | tensorflow/core/distributed_runtime/session_mgr.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/distributed_runtime/session_mgr.cc b/tensorflow/core/distributed_runtime/session_mgr.cc index 22551d5482..b97749dc41 100644 --- a/tensorflow/core/distributed_runtime/session_mgr.cc +++ b/tensorflow/core/distributed_runtime/session_mgr.cc @@ -29,7 +29,7 @@ SessionMgr::SessionMgr( std::unique_ptr<WorkerCacheInterface> default_worker_cache, WorkerCacheFactory worker_cache_factory) : worker_env_(worker_env), - legacy_session_(default_worker_name, std::move(default_worker_cache), + legacy_session_("", default_worker_name, std::move(default_worker_cache), std::unique_ptr<DeviceMgr>(worker_env->device_mgr), std::unique_ptr<GraphMgr>( new GraphMgr(worker_env, worker_env->device_mgr))), @@ -63,7 +63,7 @@ Status SessionMgr::CreateSession(const string& session, new GraphMgr(worker_env_, device_mgr.get())); std::unique_ptr<WorkerSession> worker_session(new WorkerSession( - worker_name, std::unique_ptr<WorkerCacheInterface>(worker_cache), + session, worker_name, std::unique_ptr<WorkerCacheInterface>(worker_cache), std::move(device_mgr), std::move(graph_mgr))); sessions_.insert(std::make_pair(session, std::move(worker_session))); |