Home
last modified time | relevance | path

Searched refs:worker_cache (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr.cc105 WorkerCacheInterface* worker_cache = nullptr; in CreateSession() local
108 worker_cache = new WorkerCacheWrapper(default_worker_cache_.get()); in CreateSession()
111 TF_RETURN_IF_ERROR(worker_cache_factory_(server_def, &worker_cache)); in CreateSession()
115 if (worker_cache != nullptr && default_worker_cache_ != nullptr) { in CreateSession()
116 worker_cache->SetLogging(this->is_logging_active_); in CreateSession()
155 std::unique_ptr<WorkerCacheInterface>(worker_cache), in CreateSession()
173 std::unique_ptr<WorkerCacheInterface>(worker_cache), in CreateSession()
186 void SessionMgr::ResetDefaultWorkerCache(WorkerCacheInterface* worker_cache) { in ResetDefaultWorkerCache() argument
187 default_worker_cache_.reset(worker_cache); in ResetDefaultWorkerCache()
206 WorkerCacheInterface* worker_cache = nullptr; in UpdateSession() local
[all …]
Dmaster.cc137 WorkerCacheInterface* worker_cache, in GetRemoteDevices() argument
139 DeviceFinder finder(device_filters, env, worker_cache); in GetRemoteDevices()
148 WorkerCacheInterface* worker_cache, std::vector<string>* workers) { in GetRemoteWorkers() argument
149 DeviceFinder finder(device_filters, env, worker_cache); in GetRemoteWorkers()
156 WorkerCacheInterface* worker_cache) in DeviceFinder() argument
157 : env_(env), worker_cache_(worker_cache) { in DeviceFinder()
158 CHECK(worker_cache) << "Worker cache was null!"; in DeviceFinder()
176 worker_cache->ListWorkers(&workers); in DeviceFinder()
203 worker_cache->ListWorkersInJob(job_name, &workers_in_job); in DeviceFinder()
219 worker_cache->ListWorkers(&workers); in DeviceFinder()
[all …]
Dworker_session.h49 WorkerCacheInterface* worker_cache() const { in worker_cache() function
60 std::unique_ptr<WorkerCacheInterface> worker_cache,
67 std::unique_ptr<WorkerCacheInterface> worker_cache,
92 std::unique_ptr<WorkerCacheInterface> worker_cache,
Dworker_session.cc109 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession() argument
114 worker_cache_(new WorkerFreeListCache(std::move(worker_cache))), in WorkerSession()
146 std::unique_ptr<WorkerCacheInterface> worker_cache, in CreateWithBorrowedDeviceMgr() argument
150 session_name, worker_name, std::move(worker_cache), borrowed_device_mgr, in CreateWithBorrowedDeviceMgr()
156 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession() argument
161 worker_cache_(new WorkerFreeListCache(std::move(worker_cache))), in WorkerSession()
Dremote_device.cc79 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache, in NewRemoteDevices() argument
81 WorkerInterface* wi = worker_cache->GetOrCreateWorker(worker_name); in NewRemoteDevices()
92 auto cb = [env, worker_cache, worker_name, done, wi, in NewRemoteDevices()
97 [&worker_cache, &worker_name, &wi, &done, &remote_devices, &s, call] { in NewRemoteDevices()
98 worker_cache->ReleaseWorker(worker_name, wi); in NewRemoteDevices()
Dcluster_function_library_runtime.h80 const std::shared_ptr<WorkerCacheInterface> worker_cache; member
86 std::shared_ptr<WorkerCacheInterface> worker_cache, in FunctionData()
91 worker_cache(std::move(worker_cache)), in FunctionData()
DBUILD100 ":worker_cache",
249 ":worker_cache",
275 name = "worker_cache",
276 hdrs = ["worker_cache.h"],
288 deps = [":worker_cache"],
296 ":worker_cache",
327 ":worker_cache",
350 ":worker_cache",
405 ":worker_cache",
422 ":worker_cache",
[all …]
Dcollective_rma_distributed.h32 WorkerCacheInterface* worker_cache, int64 step_id, string task_name) in CollectiveRemoteAccessDistributed() argument
34 worker_cache_(worker_cache), in CollectiveRemoteAccessDistributed()
Dcluster_function_library_runtime.cc176 worker_session_->worker_cache()->ReleaseWorker(function_data.target, in ~ClusterFunctionLibraryRuntime()
189 std::shared_ptr<WorkerCacheInterface> worker_cache = in Instantiate() local
191 WorkerInterface* wi = worker_cache->GetOrCreateWorker(target); in Instantiate()
195 worker_session_->worker_cache()->ListWorkers(&workers); in Instantiate()
237 [this, handle, req, resp, worker_cache, wi, function_name, target, in Instantiate()
243 worker_cache, wi, *send_keys, in Instantiate()
Drpc_collective_executor_mgr_test.cc45 WorkerCacheInterface* worker_cache = nullptr; in RpcCollectiveExecutorMgrTest() local
56 worker_cache, task_name)); in RpcCollectiveExecutorMgrTest()
60 MaybeCreateNcclCommunicator(), worker_cache, task_name)); in RpcCollectiveExecutorMgrTest()
Dmaster_session.cc75 bool is_partial, WorkerCacheInterface* worker_cache, in ReffedClientGraph() argument
83 worker_cache_(worker_cache), in ReffedClientGraph()
1109 WorkerCacheInterface* worker_cache = worker_cache_; in DeregisterPartitions() local
1113 auto cb = [worker_cache, c, name, w](const Status& s) { in DeregisterPartitions()
1120 worker_cache->ReleaseWorker(name, w); in DeregisterPartitions()
1230 std::unique_ptr<WorkerCacheInterface> worker_cache, in MasterSession() argument
1238 worker_cache_(std::move(worker_cache)), in MasterSession()
1292 WorkerCacheInterface* worker_cache = get_worker_cache(); in CreateWorkerSessions() local
1310 auto cleanup = gtl::MakeCleanup([&workers, worker_cache] { in CreateWorkerSessions()
1313 worker_cache->ReleaseWorker(*worker_group.name, worker_group.worker); in CreateWorkerSessions()
[all …]
Drpc_collective_executor_mgr.cc33 WorkerCacheInterface* worker_cache, const string& task_name) in RpcCollectiveExecutorMgr() argument
37 worker_cache_(worker_cache), in RpcCollectiveExecutorMgr()
Dremote_device.h60 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache,
Dmaster_env.h66 WorkerCacheInterface* worker_cache = nullptr; member
Drpc_collective_executor_mgr.h42 WorkerCacheInterface* worker_cache, const string& task_name);
Dcollective_param_resolver_distributed.h34 WorkerCacheInterface* worker_cache,
Dsession_mgr.h69 void ResetDefaultWorkerCache(WorkerCacheInterface* worker_cache);
Dsession_mgr_test.cc58 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anonc5de579e0102() argument
59 *worker_cache = nullptr; // Set to null to make debugging easier. in __anonc5de579e0102()
Dcluster_function_library_runtime_test.cc42 std::unique_ptr<WorkerCacheInterface> worker_cache( in ClusterFunctionLibraryRuntimeTest() local
47 std::move(worker_cache), std::unique_ptr<DeviceMgr>(), in ClusterFunctionLibraryRuntimeTest()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_cache_test.cc40 std::unique_ptr<WorkerCacheInterface> worker_cache( in TEST() local
43 wi = worker_cache->GetOrCreateWorker("/job:worker/replica:0/task:0"); in TEST()
45 worker_cache->ReleaseWorker("/job:worker/replica:0/task:0", wi); in TEST()
46 wi = worker_cache->GetOrCreateWorker("/job:worker/replica:0/task:1"); in TEST()
48 worker_cache->ReleaseWorker("/job:worker/replica:0/task:1", wi); in TEST()
49 wi = worker_cache->GetOrCreateWorker("/job:worker/replica:0/task:2"); in TEST()
51 worker_cache->ReleaseWorker("/job:worker/replica:0/task:2", wi); in TEST()
52 wi = worker_cache->GetOrCreateWorker("/job:worker/replica:0/task:3"); in TEST()
58 worker_cache.reset(NewGrpcWorkerCacheWithLocalWorker( in TEST()
60 wi = worker_cache->GetOrCreateWorker("local_target"); in TEST()
[all …]
Dgrpc_server_lib.cc268 WorkerCacheInterface* worker_cache; in Init() local
271 WorkerCacheFactory(worker_cache_factory_options, &worker_cache)); in Init()
272 CHECK_NE(nullptr, worker_cache); in Init()
276 opts.collective_mgr_func(config, &worker_env_, worker_cache)); in Init()
286 dev_resolver.get(), worker_cache, in Init()
290 std::move(param_resolver), MaybeCreateNcclCommunicator(), worker_cache, in Init()
297 std::unique_ptr<WorkerCacheInterface>(worker_cache), in Init()
298 [this](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in Init() argument
300 return WorkerCacheFactory(options, worker_cache); in Init()
306 master_env_.worker_cache = worker_cache; in Init()
[all …]
Drpc_rendezvous_mgr.cc119 void ReleaseWorker(WorkerCacheInterface* worker_cache) { in ReleaseWorker() argument
122 worker_cache->ReleaseWorker(src_worker_, wi_); in ReleaseWorker()
249 std::shared_ptr<WorkerCacheInterface> worker_cache = in RecvFromRemoteAsync() local
254 WorkerInterface* rwi = worker_cache->GetOrCreateWorker(call->src_worker_); in RecvFromRemoteAsync()
265 sess->worker_cache()->ReleaseWorker(call->src_worker_, rwi); in RecvFromRemoteAsync()
284 call->ReleaseWorker(sess->worker_cache()); in RecvFromRemoteAsync()
292 call->Start([this, call, worker_cache]() { in RecvFromRemoteAsync()
301 call->ReleaseWorker(session()->worker_cache()); in RecvFromRemoteAsync()
DBUILD165 "//tensorflow/core/distributed_runtime:worker_cache",
223 "//tensorflow/core/distributed_runtime:worker_cache",
302 "//tensorflow/core/distributed_runtime:worker_cache",
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext_distributed_manager.cc64 tensorflow::WorkerCacheInterface* worker_cache, in AddRemoteDevicesToMgr() argument
73 tensorflow::Env::Default(), worker_cache, added_remote_workers[i], in AddRemoteDevicesToMgr()
98 tensorflow::WorkerCacheInterface* worker_cache, in GetAllRemoteDevices() argument
101 TF_RETURN_IF_ERROR(AddRemoteDevicesToMgr(remote_workers, worker_cache, in GetAllRemoteDevices()
135 grpc_server->master_env()->worker_cache->ListWorkers(remote_workers); in ListRemoteWorkers()
459 grpc_server->master_env()->worker_cache->GetEagerClientCache( in UpdateContextWithServerDef()
490 remote_workers, grpc_server->master_env()->worker_cache, in UpdateContextWithServerDef()
534 grpc_server->master_env()->worker_cache, in UpdateContextWithServerDef()
728 grpc_server->master_env()->worker_cache->GetOrCreateWorker( in CheckRemoteAlive()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc283 worker_session->worker_cache()->ListWorkers(&remote_workers); in CreateContext()
289 TF_RETURN_IF_ERROR(worker_session->worker_cache()->GetEagerClientCache( in CreateContext()
376 worker_session->worker_cache()->ListWorkers(&remote_workers); in UpdateContext()
389 TF_RETURN_IF_ERROR(worker_session->worker_cache()->GetEagerClientCache( in UpdateContext()

12