experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Return failed status when client not found.

Browse Source 
PiperOrigin-RevId: 243105084
This commit is contained in:
Akshay Modi 2019-04-11 11:15:29 -07:00 committed by TensorFlower Gardener
parent 35e39cd3f1
commit fb8f00c0b3
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_client.cc
View File

@ -71,6 +71,10 @@ class GrpcEagerClientCache : public EagerClientCache {
if (it == clients_.end()) { if (it == clients_.end()) {
tensorflow::SharedGrpcChannelPtr shared = tensorflow::SharedGrpcChannelPtr shared =
cache_->FindWorkerChannel(target); cache_->FindWorkerChannel(target);
if (shared == nullptr) {
return errors::InvalidArgument("Client for target ", target,
" not found.");
}
auto worker = std::unique_ptr<EagerClient>(new GrpcEagerClient( auto worker = std::unique_ptr<EagerClient>(new GrpcEagerClient(
shared, threads_[AssignClientToThread(target)].completion_queue())); shared, threads_[AssignClientToThread(target)].completion_queue()));