Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_channel.cc141 SharedGrpcChannelPtr* channel_pointer) { in NewHostPortGrpcChannel()
153 SharedGrpcChannelPtr*)>& new_channel_func_ptr) { in ConvertToChannelCreationFunction()
154 return [new_channel_func_ptr](const string& target) -> SharedGrpcChannelPtr { in ConvertToChannelCreationFunction()
155 SharedGrpcChannelPtr channel_ptr; in ConvertToChannelCreationFunction()
196 SharedGrpcChannelPtr FindWorkerChannel(const string& target) override { in FindWorkerChannel()
197 SharedGrpcChannelPtr ch = nullptr; in FindWorkerChannel()
217 virtual SharedGrpcChannelPtr FindChannelOnce(const string& target) = 0;
222 std::unordered_map<string, SharedGrpcChannelPtr> channels_ TF_GUARDED_BY(mu_);
270 SharedGrpcChannelPtr FindChannelOnce(const string& target) override { in FindChannelOnce()
272 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_worker.h29 WorkerInterface* NewGrpcRemoteWorker(SharedGrpcChannelPtr channel,
Dgrpc_remote_master.cc42 explicit GrpcRemoteMaster(const SharedGrpcChannelPtr& client_channel) in GrpcRemoteMaster()
189 MasterInterface* NewGrpcMaster(const SharedGrpcChannelPtr& channel) { in NewGrpcMaster()
Dgrpc_remote_worker.cc45 explicit GrpcRemoteWorker(SharedGrpcChannelPtr channel, in GrpcRemoteWorker()
310 SharedGrpcChannelPtr channel_;
339 WorkerInterface* NewGrpcRemoteWorker(SharedGrpcChannelPtr channel, in NewGrpcRemoteWorker()
Dgrpc_util.h113 typedef std::shared_ptr<::grpc::Channel> SharedGrpcChannelPtr; typedef
Dgrpc_worker_cache.cc57 SharedGrpcChannelPtr channel = channel_cache_->FindWorkerChannel(target); in GetOrCreateWorker()
Dgrpc_channel_test.cc185 SharedGrpcChannelPtr mock_ptr; in TEST()
Dgrpc_session.cc56 SharedGrpcChannelPtr master_channel; in Create()
410 SharedGrpcChannelPtr master_channel; in Reset()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/
Dgrpc_eager_client.cc109 GrpcEagerClient(const tensorflow::SharedGrpcChannelPtr& channel, in GrpcEagerClient()
252 tensorflow::SharedGrpcChannelPtr shared = in GetClient()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc52 SharedGrpcChannelPtr channel_ptr; in MasterTest()