Home
last modified time | relevance | path

Searched defs:worker_cache (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dsession_mgr.cc119 WorkerCacheInterface* worker_cache = nullptr; in CreateSession() local
226 void SessionMgr::ResetDefaultWorkerCache(WorkerCacheInterface* worker_cache) { in ResetDefaultWorkerCache()
245 WorkerCacheInterface* worker_cache = nullptr; in UpdateSession() local
345 auto* worker_cache = legacy_session_->worker_cache(); in SetLogging() local
354 auto* worker_cache = session->worker_cache(); in SetLogging() local
366 auto* worker_cache = legacy_session_->worker_cache(); in RetrieveLogs() local
379 auto* worker_cache = session->worker_cache(); in RetrieveLogs() local
396 auto* worker_cache = legacy_session_->worker_cache(); in ClearLogs() local
405 auto* worker_cache = session->worker_cache(); in ClearLogs() local
H A Dworker_session.cc125 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
162 std::unique_ptr<WorkerCacheInterface> worker_cache, in CreateWithBorrowedDeviceMgr()
172 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
H A Dmaster.cc139 WorkerCacheInterface* worker_cache, in GetRemoteDevices()
150 WorkerCacheInterface* worker_cache, std::vector<string>* workers) { in GetRemoteWorkers()
158 WorkerCacheInterface* worker_cache) in DeviceFinder()
370 WorkerCacheInterface* worker_cache = nullptr; in CreateSession() local
H A Drpc_collective_executor_mgr.cc33 WorkerCacheInterface* worker_cache, const string& task_name) in RpcCollectiveExecutorMgr()
175 WorkerCacheInterface* worker_cache, const string& default_worker_name) { in CreateProdRpcCollectiveExecutorMgr()
H A Dmaster_session.cc81 bool is_partial, WorkerCacheInterface* worker_cache, in ReffedClientGraph()
1110 WorkerCacheInterface* worker_cache = worker_cache_; in DeregisterPartitions() local
1231 std::unique_ptr<WorkerCacheInterface> worker_cache, in MasterSession()
1292 WorkerCacheInterface* worker_cache = get_worker_cache(); in CreateWorkerSessions() local
1424 WorkerCacheInterface* worker_cache = get_worker_cache(); in DeleteWorkerSessions() local
1559 WorkerCacheInterface* worker_cache = get_worker_cache(); in StartStep() local
H A Dcollective_rma_distributed.h32 WorkerCacheInterface* worker_cache, int64_t step_id, string task_name) in CollectiveRemoteAccessDistributed()
H A Dworker_session.h49 WorkerCacheInterface* worker_cache() const { in worker_cache() function
H A Dcluster_function_library_runtime.h80 const std::shared_ptr<WorkerCacheInterface> worker_cache; member
H A Dmaster_env.h69 WorkerCacheInterface* worker_cache = nullptr; member
H A Dremote_device.cc79 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache, in NewRemoteDevices()
H A Drpc_collective_executor_mgr_test.cc45 WorkerCacheInterface* worker_cache = nullptr; in RpcCollectiveExecutorMgrTest() local
H A Dcluster_function_library_runtime_test.cc42 std::unique_ptr<WorkerCacheInterface> worker_cache( in ClusterFunctionLibraryRuntimeTest() local
H A Dsession_mgr_test.cc61 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anon955ab5140102()
H A Dcluster_function_library_runtime.cc188 std::shared_ptr<WorkerCacheInterface> worker_cache = in Instantiate() local
H A Dcollective_param_resolver_distributed.cc93 WorkerCacheInterface* worker_cache, const string& task_name) in CollectiveParamResolverDistributed()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_server_lib.cc283 WorkerCacheInterface* worker_cache; in Init() local
306 [this](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in Init()
324 std::vector<string> filtered_worker_list) { in Init()
333 WorkerCacheInterface** worker_cache) { in Init()
368 WorkerCacheInterface** worker_cache) { in WorkerCacheFactory()
461 WorkerCacheInterface* worker_cache; in UpdateServerDef() local
H A Dgrpc_worker_cache_test.cc40 std::unique_ptr<WorkerCacheInterface> worker_cache( in TEST() local
76 WorkerCacheInterface* worker_cache = in TEST() local
H A Drpc_rendezvous_mgr.cc117 void ReleaseWorker(WorkerCacheInterface* worker_cache) { in ReleaseWorker()
247 std::shared_ptr<WorkerCacheInterface> worker_cache = in RecvFromRemoteAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcontext_distributed_manager.cc74 WorkerCacheInterface* worker_cache, in AddRemoteDevicesToMgr()
106 WorkerCacheInterface* worker_cache, in GetAllRemoteDevices()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Deager_service_impl_test.cc155 WorkerCacheInterface** worker_cache) { in EagerServiceImplTest()