Home
last modified time | relevance | path

Searched refs:FindWorkerChannel (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_channel_test.cc66 EXPECT_EQ(nullptr, cc->FindWorkerChannel("invalid_target")); in TEST()
67 EXPECT_EQ(nullptr, cc->FindWorkerChannel("/job:other/replica:0/task:0")); in TEST()
68 EXPECT_EQ(nullptr, cc->FindWorkerChannel("/job:mnist/replica:0/task:6")); in TEST()
69 EXPECT_EQ(nullptr, cc->FindWorkerChannel("/job:mnist/replica:1/task:0")); in TEST()
74 auto a_1_1 = cc->FindWorkerChannel("/job:mnist/replica:0/task:0"); in TEST()
75 auto a_1_2 = cc->FindWorkerChannel("/job:mnist/replica:0/task:0"); in TEST()
77 auto d_4_1 = cc->FindWorkerChannel("/job:mnist/replica:0/task:3"); in TEST()
78 auto d_4_2 = cc->FindWorkerChannel("/job:mnist/replica:0/task:3"); in TEST()
80 auto e_5_1 = cc->FindWorkerChannel("/job:mnist/replica:0/task:4"); in TEST()
81 auto e_5_2 = cc->FindWorkerChannel("/job:mnist/replica:0/task:4"); in TEST()
[all …]
Dgrpc_channel.h77 virtual SharedGrpcChannelPtr FindWorkerChannel(const string& target) = 0;
Dgrpc_channel.cc146 SharedGrpcChannelPtr FindWorkerChannel(const string& target) override { in FindWorkerChannel() function in tensorflow::__anon5c040b370311::CachingGrpcChannelCache
222 SharedGrpcChannelPtr ch(cache->FindWorkerChannel(target)); in FindChannelOnce()
Dgrpc_worker_cache.cc76 SharedGrpcChannelPtr channel = channel_cache_->FindWorkerChannel(target); in CreateWorker()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/
Dgrpc_eager_client.cc73 cache_->FindWorkerChannel(target); in GetClient()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma_mgr.cc69 channel_cache_->FindWorkerChannel(worker_name); in SetupChannels()