| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device_test.cc | 38 std::unique_ptr<WorkerCacheInterface> worker_cache_; member in tensorflow::RemoteDeviceTest
54 worker_cache_.reset(NewGrpcWorkerCache(channel_cache)); in RemoteDeviceTest()
56 wi_ = worker_cache_->CreateWorker(remote_name_); in RemoteDeviceTest()
60 worker_cache_->ReleaseWorker(remote_name_, wi_); in ~RemoteDeviceTest()
65 NewRemoteDevices(Env::Default(), worker_cache_.get(), remote_name_, in SetUp()
|
| D | device_resolver_distributed.cc | 24 : dev_mgr_(dev_mgr), worker_cache_(worker_cache), task_name_(task_name) {} in DeviceResolverDistributed()
100 WorkerInterface* worker = worker_cache_->CreateWorker(task); in RefreshRemoteAttributes()
113 worker_cache_->ReleaseWorker(task, worker); in RefreshRemoteAttributes()
|
| D | collective_rma_distributed.h | 31 worker_cache_(worker_cache) {} in CollectiveRemoteAccessDistributed()
46 WorkerCacheInterface* worker_cache_; // Not owned
|
| D | rpc_collective_executor_mgr.cc | 35 worker_cache_(worker_cache), in RpcCollectiveExecutorMgr()
51 worker_cache_, step_id); in Create()
83 WorkerInterface* wi = worker_cache_->CreateWorker(group_leader_); in RefreshStepIdSequenceAsync()
|
| D | device_resolver_distributed.h | 60 WorkerCacheInterface* worker_cache_; // Not owned variable
|
| D | master_session.cc | 79 worker_cache_(worker_cache), in ReffedClientGraph()
103 worker_cache_->ReleaseWorker(part.name, part.worker); in ~ReffedClientGraph()
128 worker_cache_->SetLogging(active); in SetRPCLogging()
157 worker_cache_->RetrieveLogs(step_id, ss); in RetrieveLogs()
233 WorkerCacheInterface* const worker_cache_; // Not owned. member in tensorflow::MasterSession::ReffedClientGraph
446 part->worker = worker_cache_->CreateWorker(part->name); in DoRegisterPartitions()
454 worker_cache_->ReleaseWorker(part.name, part.worker); in DoRegisterPartitions()
1052 WorkerCacheInterface* worker_cache = worker_cache_; in DeregisterPartitions()
1181 worker_cache_(std::move(worker_cache)), in MasterSession()
1372 if (worker_cache_) { in ListDevices()
[all …]
|
| D | collective_param_resolver_distributed.cc | 86 worker_cache_(worker_cache), in CollectiveParamResolverDistributed()
264 cp->group, device, cancel_mgr, group_leader_, worker_cache_); in CompleteGroupDistributed()
373 group_leader_, worker_cache_); in CompleteInstanceDistributed()
|
| D | rpc_collective_executor_mgr.h | 61 WorkerCacheInterface* const worker_cache_; // Not owned. variable
|
| D | collective_param_resolver_distributed.h | 85 WorkerCacheInterface* worker_cache_; // Not owned variable
|
| 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 | collective_rma_distributed.cc | 172 &cancel_mgr_, worker_cache_)); in RecvFromPeer()
|
| D | master_session.h | 127 const std::unique_ptr<WorkerCacheInterface> worker_cache_; variable
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 73 worker_cache_(worker_cache), in CollectiveRemoteAccessDistributed()
128 &cancel_mgr_, worker_cache_)); in RecvFromPeer()
143 WorkerCacheInterface* worker_cache_; // Not owned member in tensorflow::__anon0f845a970111::CollectiveRemoteAccessDistributed
153 worker_cache_, step_id, in Create()
|