| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_cache.cc | 40 WorkerInterface* local_worker, in GrpcWorkerCache()
72 WorkerInterface* CreateWorker(const string& target) override { in CreateWorker()
84 void ReleaseWorker(const string& target, WorkerInterface* worker) override { in ReleaseWorker()
146 WorkerInterface* const local_worker_; // Not owned.
166 std::shared_ptr<GrpcChannelCache> cc, WorkerInterface* local_worker, in NewGrpcWorkerCacheWithLocalWorker()
|
| D | rpc_rendezvous_mgr.cc | 61 void Init(WorkerInterface* wi, int64 step_id, StringPiece key, in Init()
77 DCHECK_EQ(static_cast<WorkerInterface*>(nullptr), wi_) in Reset()
97 CHECK_EQ(static_cast<WorkerInterface*>(nullptr), wi_) in ~RpcRecvTensorCall()
119 DCHECK_NE(static_cast<WorkerInterface*>(nullptr), wi_) in ReleaseWorker()
156 WorkerInterface* wi_; // Not owned.
235 WorkerInterface* rwi = sess->worker_cache->CreateWorker(call->src_worker_); in RecvFromRemoteAsync()
|
| D | grpc_remote_worker.h | 30 class WorkerInterface; variable
32 WorkerInterface* NewGrpcRemoteWorker(SharedGrpcChannelPtr channel,
|
| D | grpc_worker_cache.h | 28 std::shared_ptr<GrpcChannelCache> cc, WorkerInterface* local_worker,
|
| D | grpc_remote_worker.cc | 44 class GrpcRemoteWorker : public WorkerInterface {
309 WorkerInterface* NewGrpcRemoteWorker(SharedGrpcChannelPtr channel, in NewGrpcRemoteWorker()
|
| D | rpc_rendezvous_mgr_test.cc | 55 WorkerInterface* CreateWorker(const string& target) override { in CreateWorker()
|
| /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* CreateWorker(const string& target) override { in CreateWorker()
163 void ReleaseWorker(const string& target, WorkerInterface* worker) override {} in ReleaseWorker()
187 std::unordered_map<string, WorkerInterface*> workers_;
|
| D | worker_cache.h | 49 virtual WorkerInterface* CreateWorker(const string& target) = 0;
56 virtual void ReleaseWorker(const string& target, WorkerInterface* worker) { in ReleaseWorker()
|
| D | worker_session.cc | 43 WorkerInterface* CreateWorker(const string& target) override { in CreateWorker()
57 void ReleaseWorker(const string& target, WorkerInterface* worker) override { in ReleaseWorker()
84 WorkerInterface* worker;
|
| D | worker_cache_wrapper.h | 47 virtual WorkerInterface* CreateWorker(const string& target) { in CreateWorker()
56 virtual void ReleaseWorker(const string& target, WorkerInterface* worker) { in ReleaseWorker()
|
| D | worker_cache_partial.cc | 65 auto deleter = [this, &task](WorkerInterface* wi) { in RefreshDeviceStatus()
68 std::unique_ptr<WorkerInterface, decltype(deleter)> rwi(CreateWorker(task), in RefreshDeviceStatus()
|
| D | cluster_function_library_runtime.h | 61 WorkerInterface* wi = nullptr;
66 WorkerInterface* wi, const std::vector<string>& send_keys, in FunctionData()
|
| D | worker_interface.h | 37 class WorkerInterface {
184 virtual ~WorkerInterface() {} in ~WorkerInterface()
196 typedef WorkerInterface ME;
|
| D | cancellable_call.h | 60 WorkerInterface* const wi_; // Owned by wc_, must be released.
|
| D | cluster_function_library_runtime.cc | 126 WorkerInterface* wi = in Instantiate()
177 WorkerInterface* wi = function_data->wi; in Run()
|
| D | remote_device_test.cc | 39 WorkerInterface* wi_;
|
| D | remote_device.cc | 59 WorkerInterface* wi = worker_cache->CreateWorker(worker_name); in NewRemoteDevices()
|
| D | worker.h | 42 class Worker : public WorkerInterface {
|
| D | device_resolver_distributed.cc | 100 WorkerInterface* worker = worker_cache_->CreateWorker(task); in RefreshRemoteAttributes()
|
| D | rpc_collective_executor_mgr.cc | 83 WorkerInterface* wi = worker_cache_->CreateWorker(group_leader_); in RefreshStepIdSequenceAsync()
|
| D | device_resolver_distributed_test.cc | 109 WorkerInterface* wi = it->second; in GetDeviceLocalityAsync()
|
| D | collective_rma_distributed_test.cc | 147 WorkerInterface* wi = it->second; in GetDeviceLocalityAsync()
|
| D | collective_param_resolver_distributed_test.cc | 107 WorkerInterface* wi = it->second; in GetDeviceLocalityAsync()
|
| D | master_session.cc | 260 WorkerInterface* worker = nullptr;
1054 WorkerInterface* w = part.worker; in DeregisterPartitions()
1243 WorkerInterface* worker = nullptr; in CreateWorkerSessions()
1323 WorkerInterface* worker = nullptr; in DeleteWorkerSessions()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_rendezvous_mgr.cc | 41 GdrRecvTensorCall(WorkerInterface* wi, Device* dst_device, in GdrRecvTensorCall()
107 WorkerInterface* wi_;
145 WorkerInterface* rwi = sess->worker_cache->CreateWorker(src_worker); in RecvFromRemoteAsync()
|