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

Remove another return-inside-blocking-counter pattern in context.

Browse Source 
PiperOrigin-RevId: 272059195
This commit is contained in:
Haoyu Zhang 2019-09-30 13:57:58 -07:00 committed by TensorFlower Gardener
parent 49be36bc8f
commit 11c3c50cf9
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/common_runtime/eager/context.cc
View File

@ -412,7 +412,11 @@ Status EagerContext::MaybeRegisterFunctionRemotely(const FunctionDef& fdef) {
int i = 0;
for (const auto& target : remote_contexts_) {
eager::EagerClient* eager_client;
TF_RETURN_IF_ERROR(remote_eager_workers_->GetClient(target, &eager_client));
statuses[i] = remote_eager_workers_->GetClient(target, &eager_client);
if (!statuses[i].ok()) {
blocking_counter.DecrementCount();
continue;
}
eager_client->RegisterFunctionAsync(
&request, &responses[i],