Home
last modified time | relevance | path

Searched refs:channel_cache_ (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_cache.cc44 channel_cache_(channel_cache), in GrpcWorkerCache()
64 channel_cache_->ListWorkers(workers); in ListWorkers()
69 channel_cache_->ListWorkersInJob(job_name, workers); in ListWorkersInJob()
76 SharedGrpcChannelPtr channel = channel_cache_->FindWorkerChannel(target); in CreateWorker()
147 std::shared_ptr<GrpcChannelCache> channel_cache_; member in tensorflow::__anonab8e586b0111::GrpcWorkerCache
Dgrpc_server_lib.h98 std::shared_ptr<GrpcChannelCache> channel_cache() { return channel_cache_; } in channel_cache()
153 std::shared_ptr<GrpcChannelCache> channel_cache_; variable
Dgrpc_server_lib.cc319 channel_cache_.reset( in WorkerCacheFactory()
325 const string host_port = channel_cache_->TranslateTask(name_prefix); in WorkerCacheFactory()
341 channel_cache_, worker_impl_.get(), name_prefix); in WorkerCacheFactory()
/external/tensorflow/tensorflow/contrib/verbs/
Dverbs_server_lib.cc48 delete channel_cache_; in ~VerbsServer()
91 CHECK(ChannelCacheFactory(server_def(), &channel_cache_).ok()); in Init()
92 rdma_mgr_ = new RdmaMgr(worker_env(), channel_cache_); in Init()
Drdma_mgr.h49 GrpcChannelCache* const channel_cache_; variable
Dverbs_server_lib.h60 GrpcChannelCache* channel_cache_ = nullptr; variable
Drdma_mgr.cc37 : worker_env_(worker_env), channel_cache_(channel_cache) { in RdmaMgr()
69 channel_cache_->FindWorkerChannel(worker_name); in SetupChannels()