| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device_test.cc | 38 std::unique_ptr<WorkerCacheInterface> worker_cache_; member in tensorflow::RemoteDeviceTest
56 worker_cache_.reset( in RemoteDeviceTest()
59 wi_ = worker_cache_->GetOrCreateWorker(remote_name_); in RemoteDeviceTest()
63 worker_cache_->ReleaseWorker(remote_name_, wi_); in ~RemoteDeviceTest()
68 NewRemoteDevices(Env::Default(), worker_cache_.get(), remote_name_, in SetUp()
|
| D | worker_session.h | 51 return worker_cache_.get(); in worker_cache()
77 return worker_cache_; in GetSharedWorkerCache()
105 std::shared_ptr<WorkerCacheInterface> worker_cache_
|
| D | collective_rma_distributed.h | 34 worker_cache_(worker_cache), in CollectiveRemoteAccessDistributed()
55 WorkerCacheInterface* worker_cache_; // Not owned
|
| D | rpc_collective_executor_mgr.cc | 37 worker_cache_(worker_cache), in RpcCollectiveExecutorMgr()
53 work_queue_, worker_cache_, step_id, in Create()
86 WorkerInterface* wi = worker_cache_->GetOrCreateWorker(group_leader_); in RefreshStepIdSequenceAsync()
|
| D | worker_session.cc | 114 worker_cache_(new WorkerFreeListCache(std::move(worker_cache))), in WorkerSession()
134 worker_cache_ = std::shared_ptr<WorkerCacheInterface>( in UpdateWorkerCacheAndDevices()
161 worker_cache_(new WorkerFreeListCache(std::move(worker_cache))), in WorkerSession()
|
| D | collective_rma_distributed.cc | 174 cancellation_manager, worker_cache_)); in RecvFromPeer()
201 WorkerInterface* wi = worker_cache_->GetOrCreateWorker(peer_task); in CheckPeerHealth()
245 worker_cache_->ReleaseWorker(peer_task, wi); in CheckPeerHealth()
|
| D | rpc_collective_executor_mgr.h | 62 WorkerCacheInterface* const worker_cache_; // Not owned. variable
|
| D | collective_param_resolver_distributed.h | 91 WorkerCacheInterface* worker_cache_; // Not owned variable
|
| D | master_session.cc | 83 worker_cache_(worker_cache), in ReffedClientGraph()
107 worker_cache_->ReleaseWorker(part.name, part.worker); in ~ReffedClientGraph()
132 worker_cache_->SetLogging(active); in SetRPCLogging()
161 worker_cache_->RetrieveLogs(step_id, ss); in RetrieveLogs()
237 WorkerCacheInterface* const worker_cache_; // Not owned. member in tensorflow::MasterSession::ReffedClientGraph
450 part->worker = worker_cache_->GetOrCreateWorker(part->name); in DoRegisterPartitions()
458 worker_cache_->ReleaseWorker(part.name, part.worker); in DoRegisterPartitions()
1109 WorkerCacheInterface* worker_cache = worker_cache_; in DeregisterPartitions()
1238 worker_cache_(std::move(worker_cache)), in MasterSession()
1464 if (worker_cache_) { in ListDevices()
[all …]
|
| D | collective_param_resolver_distributed.cc | 96 worker_cache_(worker_cache), in CollectiveParamResolverDistributed()
299 group_leader_, worker_cache_); in CompleteGroupDistributed()
389 group_leader_, worker_cache_); in CompleteInstanceDistributed()
|
| D | master.cc | 157 : env_(env), worker_cache_(worker_cache) { in DeviceFinder()
249 NewRemoteDevices(env_->env, worker_cache_, targets_[i], in Start()
297 WorkerCacheInterface* worker_cache_; member in tensorflow::DeviceFinder
|
| D | master_session.h | 125 const std::unique_ptr<WorkerCacheInterface> worker_cache_; variable
|