| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_cache.cc | 35 WorkerInterface* local_worker, in GrpcWorkerCache()
53 WorkerInterface* GetOrCreateWorker(const string& target) override { in GetOrCreateWorker()
68 void ReleaseWorker(const string& target, WorkerInterface* worker) override { in ReleaseWorker()
108 WorkerInterface* const local_worker_; // Not owned.
171 WorkerInterface* local_worker, const string& local_target) { in NewGrpcWorkerCacheWithLocalWorker()
|
| D | rpc_rendezvous_mgr.cc | 62 void Init(WorkerInterface* wi, int64 step_id, StringPiece key, in Init()
78 DCHECK_EQ(static_cast<WorkerInterface*>(nullptr), wi_) in Reset()
98 CHECK_EQ(static_cast<WorkerInterface*>(nullptr), wi_) in ~RpcRecvTensorCall()
120 DCHECK_NE(static_cast<WorkerInterface*>(nullptr), wi_) in ReleaseWorker()
173 WorkerInterface* wi_; // Not owned.
254 WorkerInterface* rwi = worker_cache->GetOrCreateWorker(call->src_worker_); in RecvFromRemoteAsync()
|
| D | grpc_remote_worker.h | 27 class WorkerInterface; variable
29 WorkerInterface* NewGrpcRemoteWorker(SharedGrpcChannelPtr channel,
|
| D | grpc_worker_cache.h | 73 WorkerInterface* local_worker, const string& local_target);
|
| D | grpc_worker_cache_test.cc | 42 WorkerInterface* wi; in TEST()
|
| D | grpc_remote_worker.cc | 43 class GrpcRemoteWorker : public WorkerInterface {
339 WorkerInterface* NewGrpcRemoteWorker(SharedGrpcChannelPtr channel, in NewGrpcRemoteWorker()
|
| D | rpc_rendezvous_mgr_test.cc | 76 WorkerInterface* GetOrCreateWorker(const string& target) override { in GetOrCreateWorker()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | test_utils.h | 31 class TestWorkerInterface : public WorkerInterface {
127 void AddWorker(const string& target, WorkerInterface* wi) { in AddWorker()
155 WorkerInterface* GetOrCreateWorker(const string& target) override { in GetOrCreateWorker()
163 void ReleaseWorker(const string& target, WorkerInterface* worker) override {} in ReleaseWorker()
192 std::unordered_map<string, WorkerInterface*> workers_;
|
| D | worker_cache.h | 47 virtual WorkerInterface* GetOrCreateWorker(const string& target) = 0;
54 virtual void ReleaseWorker(const string& target, WorkerInterface* worker) { in ReleaseWorker()
|
| D | worker_cache_wrapper.h | 44 WorkerInterface* GetOrCreateWorker(const string& target) override { in GetOrCreateWorker()
53 void ReleaseWorker(const string& target, WorkerInterface* worker) override { in ReleaseWorker()
|
| D | worker_cache_partial.cc | 65 auto deleter = [this, &task](WorkerInterface* wi) { in RefreshDeviceStatus()
68 std::unique_ptr<WorkerInterface, decltype(deleter)> rwi( in RefreshDeviceStatus()
|
| D | worker_session.cc | 50 WorkerInterface* GetOrCreateWorker(const string& target) override { in GetOrCreateWorker()
69 void ReleaseWorker(const string& target, WorkerInterface* worker) override { in ReleaseWorker()
96 WorkerInterface* worker;
|
| D | worker_interface.h | 37 class WorkerInterface {
192 virtual ~WorkerInterface() {} in ~WorkerInterface()
204 typedef WorkerInterface ME;
|
| D | cluster_function_library_runtime.h | 81 WorkerInterface* wi = nullptr;
87 WorkerInterface* wi, const std::vector<string>& send_keys, in FunctionData()
|
| D | cancellable_call.h | 56 WorkerInterface* const wi_; // Owned by wc_, must be released.
|
| D | cluster_function_library_runtime.cc | 191 WorkerInterface* wi = worker_cache->GetOrCreateWorker(target); in Instantiate()
268 WorkerInterface* wi = function_data->wi; in Run()
372 WorkerInterface* wi = function_data->wi; in CleanUp()
|
| D | remote_device_test.cc | 39 WorkerInterface* wi_;
|
| D | worker.h | 43 class Worker : public WorkerInterface {
|
| D | remote_device.cc | 81 WorkerInterface* wi = worker_cache->GetOrCreateWorker(worker_name); in NewRemoteDevices()
|
| D | rpc_collective_executor_mgr.cc | 86 WorkerInterface* wi = worker_cache_->GetOrCreateWorker(group_leader_); in RefreshStepIdSequenceAsync()
|
| D | collective_rma_distributed.cc | 201 WorkerInterface* wi = worker_cache_->GetOrCreateWorker(peer_task); in CheckPeerHealth()
|
| D | collective_param_resolver_distributed_test.cc | 110 WorkerInterface* wi = it->second; in GetDeviceLocalityAsync()
|
| D | master_session.cc | 264 WorkerInterface* worker = nullptr;
1111 WorkerInterface* w = part.worker; in DeregisterPartitions()
1299 WorkerInterface* worker = nullptr; in CreateWorkerSessions()
1415 WorkerInterface* worker = nullptr; in DeleteWorkerSessions()
|
| D | collective_rma_distributed_test.cc | 163 WorkerInterface* wi = it->second; in GetDeviceLocalityAsync()
|
| D | message_wrappers.h | 515 friend class WorkerInterface; variable
|