Home
last modified time | relevance | path

Searched refs:SharedGrpcChannelPtr (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_channel.cc91 SharedGrpcChannelPtr* channel_pointer) { in NewHostPortGrpcChannel()
103 SharedGrpcChannelPtr*)>& new_channel_func_ptr) { in ConvertToChannelCreationFunction()
104 return [new_channel_func_ptr](const string& target) -> SharedGrpcChannelPtr { in ConvertToChannelCreationFunction()
105 SharedGrpcChannelPtr channel_ptr; in ConvertToChannelCreationFunction()
146 SharedGrpcChannelPtr FindWorkerChannel(const string& target) override { in FindWorkerChannel()
147 SharedGrpcChannelPtr ch = nullptr; in FindWorkerChannel()
167 virtual SharedGrpcChannelPtr FindChannelOnce(const string& target) = 0;
172 std::unordered_map<string, SharedGrpcChannelPtr> channels_ GUARDED_BY(mu_);
220 SharedGrpcChannelPtr FindChannelOnce(const string& target) override { in FindChannelOnce()
222 SharedGrpcChannelPtr ch(cache->FindWorkerChannel(target)); in FindChannelOnce()
[all …]
Dgrpc_channel.h77 virtual SharedGrpcChannelPtr FindWorkerChannel(const string& target) = 0;
83 typedef std::function<SharedGrpcChannelPtr(string)> ChannelCreationFunction;
94 SharedGrpcChannelPtr*)>& new_channel_func_ptr);
98 SharedGrpcChannelPtr* channel_pointer);
Dgrpc_remote_master.h24 MasterInterface* NewGrpcMaster(const SharedGrpcChannelPtr& channel);
Dgrpc_remote_master.cc39 explicit GrpcRemoteMaster(const SharedGrpcChannelPtr& client_channel) in GrpcRemoteMaster()
185 MasterInterface* NewGrpcMaster(const SharedGrpcChannelPtr& channel) { in NewGrpcMaster()
Dgrpc_remote_worker.h32 WorkerInterface* NewGrpcRemoteWorker(SharedGrpcChannelPtr channel,
Dgrpc_remote_worker.cc46 explicit GrpcRemoteWorker(SharedGrpcChannelPtr channel, in GrpcRemoteWorker()
282 SharedGrpcChannelPtr channel_;
309 WorkerInterface* NewGrpcRemoteWorker(SharedGrpcChannelPtr channel, in NewGrpcRemoteWorker()
Dgrpc_util.h113 typedef std::shared_ptr<::grpc::Channel> SharedGrpcChannelPtr; typedef
Dgrpc_worker_cache.cc76 SharedGrpcChannelPtr channel = channel_cache_->FindWorkerChannel(target); in CreateWorker()
Dgrpc_channel_test.cc185 SharedGrpcChannelPtr mock_ptr; in TEST()
Dgrpc_session.cc55 SharedGrpcChannelPtr master_channel; in Create()
393 SharedGrpcChannelPtr master_channel; in Reset()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/
Dgrpc_eager_client.cc32 GrpcEagerClient(const tensorflow::SharedGrpcChannelPtr& channel, in GrpcEagerClient()
72 tensorflow::SharedGrpcChannelPtr shared = in GetClient()
/external/tensorflow/tensorflow/contrib/verbs/
Dgrpc_verbs_client.h30 explicit GrpcVerbsClient(SharedGrpcChannelPtr client_channel) in GrpcVerbsClient()
Drdma_mgr.cc68 SharedGrpcChannelPtr client_channel = in SetupChannels()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc52 SharedGrpcChannelPtr channel_ptr; in MasterTest()