Home
last modified time | relevance | path

Searched refs:ReleaseWorker (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_cache_test.cc45 worker_cache->ReleaseWorker("/job:worker/replica:0/task:0", wi); in TEST()
48 worker_cache->ReleaseWorker("/job:worker/replica:0/task:1", wi); in TEST()
51 worker_cache->ReleaseWorker("/job:worker/replica:0/task:2", wi); in TEST()
Drpc_rendezvous_mgr.cc119 void ReleaseWorker(WorkerCacheInterface* worker_cache) { in ReleaseWorker() function in tensorflow::__anona9090a8f0111::RpcRecvTensorCall
122 worker_cache->ReleaseWorker(src_worker_, wi_); in ReleaseWorker()
265 sess->worker_cache()->ReleaseWorker(call->src_worker_, rwi); in RecvFromRemoteAsync()
284 call->ReleaseWorker(sess->worker_cache()); in RecvFromRemoteAsync()
301 call->ReleaseWorker(session()->worker_cache()); in RecvFromRemoteAsync()
Dgrpc_worker_cache.cc68 void ReleaseWorker(const string& target, WorkerInterface* worker) override { in ReleaseWorker() function in tensorflow::__anon75b4e2090111::GrpcWorkerCache
73 WorkerCacheInterface::ReleaseWorker(target, worker); in ReleaseWorker()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_cache_wrapper.h53 void ReleaseWorker(const string& target, WorkerInterface* worker) override { in ReleaseWorker() function
54 return wrapped_->ReleaseWorker(target, worker); in ReleaseWorker()
Dcancellable_call.h38 virtual ~CancellableCall() { wc_->ReleaseWorker(remote_worker_, wi_); } in ~CancellableCall()
Dworker_session.cc37 wrapped_->ReleaseWorker(p.first, p.second.worker); in ~WorkerFreeListCache()
69 void ReleaseWorker(const string& target, WorkerInterface* worker) override { in ReleaseWorker() function in tensorflow::__anonf821b21a0111::WorkerFreeListCache
Dworker_cache.h54 virtual void ReleaseWorker(const string& target, WorkerInterface* worker) { in ReleaseWorker() function
Dworker_cache_partial.cc66 ReleaseWorker(task, wi); in RefreshDeviceStatus()
Dremote_device_test.cc63 worker_cache_->ReleaseWorker(remote_name_, wi_); in ~RemoteDeviceTest()
Dremote_device.cc98 worker_cache->ReleaseWorker(worker_name, wi); in NewRemoteDevices()
Dtest_utils.h163 void ReleaseWorker(const string& target, WorkerInterface* worker) override {} in ReleaseWorker() function
Dcollective_rma_distributed.cc245 worker_cache_->ReleaseWorker(peer_task, wi); in CheckPeerHealth()
Dmaster_session.cc107 worker_cache_->ReleaseWorker(part.name, part.worker); in ~ReffedClientGraph()
458 worker_cache_->ReleaseWorker(part.name, part.worker); in DoRegisterPartitions()
1120 worker_cache->ReleaseWorker(name, w); in DeregisterPartitions()
1313 worker_cache->ReleaseWorker(*worker_group.name, worker_group.worker); in CreateWorkerSessions()
1431 worker_cache->ReleaseWorker(*worker_group.name, worker_group.worker); in DeleteWorkerSessions()
Dcluster_function_library_runtime.cc176 worker_session_->worker_cache()->ReleaseWorker(function_data.target, in ~ClusterFunctionLibraryRuntime()
Dmaster.cc637 env_->worker_cache->ReleaseWorker(worker_name, worker); in CleanupWorkers()