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

Fix Kokoro build.

Browse Source 
PiperOrigin-RevId: 283881484
Change-Id: I13edeb9da81c385eea9b7b402a572f2501f3ce6b
This commit is contained in:
Haoyu Zhang 2019-12-04 18:26:01 -08:00 committed by TensorFlower Gardener
parent b21d30f6aa
commit 7a33433911
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -230,8 +230,9 @@ class GrpcEagerClientCache : public EagerClientCache {
} }
int assigned_index = AssignClientToThread(target); int assigned_index = AssignClientToThread(target);
GrpcEagerClientThread* thread = threads_[assigned_index].get(); GrpcEagerClientThread* thread = threads_[assigned_index].get();
auto worker = new GrpcEagerClient(shared, thread); core::RefCountPtr<EagerClient> worker(
it = clients_.emplace(target, worker).first; new GrpcEagerClient(shared, thread));
it = clients_.emplace(target, std::move(worker)).first;
} }
it->second->Ref(); it->second->Ref();