Searched refs:worker_session (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.cc | 77 std::shared_ptr<WorkerSession> worker_session; in CreateSession() local
89 worker_session.reset( in CreateSession()
98 worker_session = WorkerSession::CreateWithBorrowedDeviceMgr( in CreateSession()
104 sessions_.insert(std::make_pair(session, std::move(worker_session))); in CreateSession()
|
| D | cluster_function_library_runtime.h | 30 ClusterFunctionLibraryRuntime(WorkerSession* worker_session, in ClusterFunctionLibraryRuntime() argument
32 : worker_session_(worker_session), in ClusterFunctionLibraryRuntime()
|
| D | BUILD | 86 name = "worker_session",
89 "worker_session.cc",
93 "worker_session.h",
111 ":worker_session",
131 ":worker_session",
204 ":worker_session",
396 ":worker_session",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 113 std::shared_ptr<WorkerSession> worker_session; in CreateContext() local
115 session_name, &worker_session)); in CreateContext()
118 TF_RETURN_IF_ERROR(r->Initialize(worker_session.get())); in CreateContext()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | BUILD | 71 "//tensorflow/core/distributed_runtime:worker_session",
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 219 std::shared_ptr<tensorflow::WorkerSession> worker_session; in UpdateTFE_ContextWithServerDef() local
222 session_name, &worker_session)); in UpdateTFE_ContextWithServerDef()
225 TF_RETURN_IF_ERROR(r->Initialize(worker_session.get())); in UpdateTFE_ContextWithServerDef()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | BUILD | 69 "//tensorflow/core/distributed_runtime:worker_session",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | BUILD | 197 "//tensorflow/core/distributed_runtime:worker_session",
|