| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.cc | 95 WorkerSession::WorkerSession(const string& session_name, in WorkerSession() function in tensorflow::WorkerSession
110 std::shared_ptr<WorkerSession> WorkerSession::CreateWithBorrowedDeviceMgr( in CreateWithBorrowedDeviceMgr()
114 return std::shared_ptr<WorkerSession>( in CreateWithBorrowedDeviceMgr()
115 new WorkerSession(session_name, worker_name, std::move(worker_cache), in CreateWithBorrowedDeviceMgr()
119 WorkerSession::WorkerSession(const string& session_name, in WorkerSession() function in tensorflow::WorkerSession
133 WorkerSession::~WorkerSession() { in ~WorkerSession()
|
| D | session_mgr_test.cc | 72 std::shared_ptr<WorkerSession> session; in TEST_F()
89 std::shared_ptr<WorkerSession> session; in TEST_F()
100 std::shared_ptr<WorkerSession> session; in TEST_F()
113 std::shared_ptr<WorkerSession> session_1; in TEST_F()
119 std::shared_ptr<WorkerSession> session_2; in TEST_F()
125 std::shared_ptr<WorkerSession> session_3; in TEST_F()
131 std::shared_ptr<WorkerSession> session_4; in TEST_F()
144 std::shared_ptr<WorkerSession> session; in TEST_F()
153 std::shared_ptr<WorkerSession> session; in TEST_F()
|
| D | worker_session.h | 33 struct WorkerSession { struct
60 WorkerSession(const string& session_name, const string& worker_name, argument
65 static std::shared_ptr<WorkerSession> CreateWithBorrowedDeviceMgr(
70 ~WorkerSession();
73 WorkerSession(const string& session_name, const string& worker_name,
|
| D | session_mgr.h | 54 std::shared_ptr<WorkerSession>* out_session);
55 std::shared_ptr<WorkerSession> LegacySession();
84 std::shared_ptr<WorkerSession> legacy_session_;
91 const string& session_handle, std::shared_ptr<WorkerSession>* out_session)
96 std::map<string, std::shared_ptr<WorkerSession>> sessions_ GUARDED_BY(mu_);
|
| D | session_mgr.cc | 37 legacy_session_(WorkerSession::CreateWithBorrowedDeviceMgr( in SessionMgr()
77 std::shared_ptr<WorkerSession> worker_session; in CreateSession()
90 new WorkerSession(session, worker_name, in CreateSession()
98 worker_session = WorkerSession::CreateWithBorrowedDeviceMgr( in CreateSession()
118 const string& session_handle, std::shared_ptr<WorkerSession>* out_session) { in WorkerSessionForSessionLocked()
136 const string& session_handle, std::shared_ptr<WorkerSession>* out_session) { in WorkerSessionForSession()
141 std::shared_ptr<WorkerSession> SessionMgr::LegacySession() { in LegacySession()
|
| D | cluster_function_library_runtime.h | 24 struct WorkerSession;
30 ClusterFunctionLibraryRuntime(WorkerSession* worker_session, in ClusterFunctionLibraryRuntime()
55 WorkerSession* const worker_session_ = nullptr; // not owned.
|
| D | rendezvous_mgr_interface.h | 28 struct WorkerSession;
42 virtual Status Initialize(WorkerSession* session) = 0;
|
| D | base_rendezvous_mgr.h | 124 Status Initialize(WorkerSession* session) override;
168 WorkerSession* session();
184 WorkerSession* session_ GUARDED_BY(mu_); // Not owned.
|
| D | graph_mgr.h | 46 struct WorkerSession;
90 WorkerSession* session, const ExecutorOpts& opts,
|
| D | base_rendezvous_mgr.cc | 152 Status BaseRemoteRendezvous::Initialize(WorkerSession* session) { in Initialize()
177 WorkerSession* BaseRemoteRendezvous::session() { in session()
209 WorkerSession* sess = nullptr; in ValidateDevices()
254 WorkerSession* sess = session(); in SameWorkerRecvDone()
|
| D | worker.cc | 65 std::shared_ptr<WorkerSession> session; in RegisterGraphAsync()
86 std::shared_ptr<WorkerSession> session; in DeregisterGraphAsync()
159 std::shared_ptr<WorkerSession> session; in DoRunGraph()
269 std::shared_ptr<WorkerSession> session; in DoPartialRunGraph()
|
| D | cluster_function_library_runtime_test.cc | 42 worker_session_.reset(new WorkerSession( in ClusterFunctionLibraryRuntimeTest()
100 std::unique_ptr<WorkerSession> worker_session_;
|
| D | worker.h | 31 struct WorkerSession;
|
| D | graph_mgr.cc | 392 WorkerSession* session, const ExecutorOpts& opts, in ExecuteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.h | 36 struct WorkerSession;
|
| D | grpc_server_lib.h | 164 std::shared_ptr<WorkerSession> worker_session_;
|
| D | rpc_rendezvous_mgr_test.cc | 82 WorkerSession worker_session_;
|
| D | rpc_rendezvous_mgr.cc | 231 WorkerSession* sess = session(); in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_rendezvous_mgr.cc | 144 WorkerSession* sess = session(); in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 113 std::shared_ptr<WorkerSession> worker_session; in CreateContext()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 219 std::shared_ptr<tensorflow::WorkerSession> worker_session; in UpdateTFE_ContextWithServerDef()
|