Home
last modified time | relevance | path

Searched defs:worker_session (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dsession_mgr.cc136 std::shared_ptr<WorkerSession> worker_session; in CreateSession() local
243 std::shared_ptr<WorkerSession> worker_session = it->second; in UpdateSession() local
H A Dcluster_function_library_runtime.h31 ClusterFunctionLibraryRuntime(WorkerSession* worker_session, in ClusterFunctionLibraryRuntime()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Deager_service_impl.cc259 std::shared_ptr<WorkerSession> worker_session; in CreateContext() local
403 std::shared_ptr<WorkerSession> worker_session; in UpdateContext() local
H A Dcluster_function_library_runtime.cc287 const uint64 context_id, EagerContext* ctx, WorkerSession* worker_session) { in CreateClusterFLR()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcontext_distributed_manager.cc605 std::shared_ptr<WorkerSession> worker_session; in UpdateContextWithServerDef() local
709 std::shared_ptr<WorkerSession> worker_session; in EnableCollectiveOps() local
H A Dcontext.cc1452 std::shared_ptr<WorkerSession> worker_session) { in SetWorkerEnv()
1459 std::shared_ptr<WorkerSession> worker_session, in InitializeRemoteMaster()
1557 std::shared_ptr<WorkerSession> worker_session, in SetMasterContextState()