Home
last modified time | relevance | path

Searched refs:FindWorkerChannel (Results 1 – 5 of 5) 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.cc196 SharedGrpcChannelPtr FindWorkerChannel(const string& target) override { in FindWorkerChannel() function in tensorflow::__anon3b8f1f390311::CachingGrpcChannelCache
272 SharedGrpcChannelPtr ch(cache->FindWorkerChannel(target)); in FindChannelOnce()
Dgrpc_worker_cache.cc67 SharedGrpcChannelPtr channel = channel_cache_->FindWorkerChannel(target); in GetOrCreateWorker()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/
Dgrpc_eager_client.cc226 cache_->FindWorkerChannel(target); in GetClient()