| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.cc | 107 WorkerSession::WorkerSession( in WorkerSession() function in tensorflow::WorkerSession
128 Status WorkerSession::UpdateWorkerCacheAndDevices( in UpdateWorkerCacheAndDevices()
144 std::shared_ptr<WorkerSession> WorkerSession::CreateWithBorrowedDeviceMgr( in CreateWithBorrowedDeviceMgr()
149 return std::shared_ptr<WorkerSession>(new WorkerSession( in CreateWithBorrowedDeviceMgr()
154 WorkerSession::WorkerSession( in WorkerSession() function in tensorflow::WorkerSession
174 WorkerSession::~WorkerSession() { in ~WorkerSession()
|
| D | session_mgr_test.cc | 72 std::shared_ptr<WorkerSession> session; in TEST_F()
96 std::shared_ptr<WorkerSession> session; in TEST_F()
113 std::shared_ptr<WorkerSession> session; in TEST_F()
126 std::shared_ptr<WorkerSession> session_1; in TEST_F()
132 std::shared_ptr<WorkerSession> session_2; in TEST_F()
138 std::shared_ptr<WorkerSession> session_3; in TEST_F()
144 std::shared_ptr<WorkerSession> session_4; in TEST_F()
179 std::shared_ptr<WorkerSession> session; in TEST_F()
228 std::shared_ptr<WorkerSession> session; in TEST_F()
241 std::shared_ptr<WorkerSession> session; in TEST_F()
[all …]
|
| D | session_mgr.h | 80 std::shared_ptr<WorkerSession>* out_session);
81 std::shared_ptr<WorkerSession> LegacySession();
110 std::shared_ptr<WorkerSession> legacy_session_;
117 const string& session_handle, std::shared_ptr<WorkerSession>* out_session)
122 std::map<string, std::shared_ptr<WorkerSession>> sessions_ TF_GUARDED_BY(mu_);
|
| D | worker_session.h | 33 class WorkerSession {
59 WorkerSession(const string& session_name, const string& worker_name,
65 static std::shared_ptr<WorkerSession> CreateWithBorrowedDeviceMgr(
88 ~WorkerSession();
91 WorkerSession(const string& session_name, const string& worker_name,
|
| D | graph_mgr.h | 46 class WorkerSession; variable
78 WorkerSession* session, const GraphOptions& graph_options,
91 WorkerSession* session, const ExecutorOpts& opts,
172 WorkerSession* session, StatusCallback done);
182 WorkerSession* session, const GraphOptions& graph_options,
|
| D | session_mgr.cc | 38 legacy_session_(WorkerSession::CreateWithBorrowedDeviceMgr( in SessionMgr()
122 std::shared_ptr<WorkerSession> worker_session; in CreateSession()
154 new WorkerSession(session, worker_name, in CreateSession()
171 worker_session = WorkerSession::CreateWithBorrowedDeviceMgr( in CreateSession()
204 std::shared_ptr<WorkerSession> worker_session = it->second; in UpdateSession()
264 const string& session_handle, std::shared_ptr<WorkerSession>* out_session) { in WorkerSessionForSessionLocked()
282 const string& session_handle, std::shared_ptr<WorkerSession>* out_session) { in WorkerSessionForSession()
287 std::shared_ptr<WorkerSession> SessionMgr::LegacySession() { in LegacySession()
|
| D | cluster_function_library_runtime.h | 25 class WorkerSession; variable
31 ClusterFunctionLibraryRuntime(WorkerSession* worker_session, in ClusterFunctionLibraryRuntime()
70 WorkerSession* const worker_session_ = nullptr; // not owned.
|
| D | rendezvous_mgr_interface.h | 28 class WorkerSession; variable
42 virtual Status Initialize(WorkerSession* session) = 0;
|
| D | base_rendezvous_mgr.h | 121 Status Initialize(WorkerSession* session) override;
165 WorkerSession* session();
182 WorkerSession* session_ TF_GUARDED_BY(mu_); // Not owned.
|
| D | base_rendezvous_mgr.cc | 138 Status BaseRemoteRendezvous::Initialize(WorkerSession* session) { in Initialize()
163 WorkerSession* BaseRemoteRendezvous::session() { in session()
177 WorkerSession* sess = nullptr; in Send()
199 WorkerSession* sess = nullptr; in ValidateDevices()
248 WorkerSession* sess = session(); in SameWorkerRecvDone()
|
| D | worker.cc | 72 std::shared_ptr<WorkerSession> session; in RegisterGraphAsync()
93 std::shared_ptr<WorkerSession> session; in DeregisterGraphAsync()
173 std::shared_ptr<WorkerSession> session; in DoRunGraph()
275 std::shared_ptr<WorkerSession> session; in DoPartialRunGraph()
|
| D | graph_mgr.cc | 126 const string& handle, const GraphDef& gdef, WorkerSession* session, in InitItem()
296 const string& handle, const GraphDef& gdef, WorkerSession* session, in Register()
416 WorkerSession* session, const ExecutorOpts& opts, in ExecuteAsync()
515 WorkerSession* session, StatusCallback done) { in StartParallelExecutors()
|
| D | cluster_function_library_runtime_test.cc | 45 worker_session_.reset(new WorkerSession( in ClusterFunctionLibraryRuntimeTest()
113 std::unique_ptr<WorkerSession> worker_session_;
|
| D | worker.h | 32 class WorkerSession; variable
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.h | 29 class WorkerSession; variable
108 const uint64 context_id, EagerContext* ctx, WorkerSession* worker_session);
|
| D | cluster_function_library_runtime.cc | 286 const uint64 context_id, EagerContext* ctx, WorkerSession* worker_session) { in CreateClusterFLR()
|
| D | eager_service_impl.cc | 254 std::shared_ptr<WorkerSession> worker_session; in CreateContext()
369 std::shared_ptr<WorkerSession> worker_session; in UpdateContext()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.h | 36 class WorkerSession; variable
|
| D | grpc_server_lib.h | 187 std::shared_ptr<WorkerSession> worker_session_;
|
| D | rpc_rendezvous_mgr.cc | 248 WorkerSession* sess = session(); in RecvFromRemoteAsync()
|
| D | rpc_rendezvous_mgr_test.cc | 135 WorkerSession worker_session_;
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 367 std::shared_ptr<WorkerSession> worker_session,
669 std::shared_ptr<WorkerSession> worker_session,
683 std::shared_ptr<WorkerSession> worker_session_;
|
| D | context.cc | 1202 std::shared_ptr<WorkerSession> worker_session, in InitializeRemoteMaster()
1300 std::shared_ptr<WorkerSession> worker_session, in SetMasterContextState()
|
| D | context_distributed_manager.cc | 609 std::shared_ptr<tensorflow::WorkerSession> worker_session; in UpdateContextWithServerDef()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 78 // If the target worker already has a WorkerSession created previously with
80 // that the previous master failed before deleting the WorkerSession on the
|