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

Clear cancel callback when gRPC eager call returns with state.

Browse Source 
PiperOrigin-RevId: 317393892
Change-Id: Ife800821494dd4cc2992eec9a5470d989596a6d7
This commit is contained in:
Haoyu Zhang 2020-06-19 15:43:22 -07:00 committed by TensorFlower Gardener
parent fb5a3c3c6a
commit 18c43b7bd4
1 changed files with 1 additions and 0 deletions

1
tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.h
View File

@ -81,6 +81,7 @@ class GrpcEagerServiceImpl : public AsyncServiceInterface {
local_impl_.RunComponentFunction(call_opts.get(), &call->request, local_impl_.RunComponentFunction(call_opts.get(), &call->request,
&call->response, &call->response,
[call, call_opts](const Status& s) { [call, call_opts](const Status& s) {
call->ClearCancelCallback();
call->SendResponse(ToGrpcStatus(s)); call->SendResponse(ToGrpcStatus(s));
}); });
}); });