diff options
Diffstat (limited to 'tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.h')
-rw-r--r-- | tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.h b/tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.h index 9a94026342..537e9043bd 100644 --- a/tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.h +++ b/tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.h @@ -45,7 +45,7 @@ class GrpcEagerServiceImpl : public AsyncServiceInterface { private: #define HANDLER(method) \ void method##Handler(EagerCall<method##Request, method##Response>* call) { \ - request_handler_threadpool_->Schedule([this, call]() { \ + env_->compute_pool->Schedule([this, call]() { \ call->SendResponse( \ ToGrpcStatus(local_impl_.method(&call->request, &call->response))); \ }); \ @@ -62,8 +62,10 @@ class GrpcEagerServiceImpl : public AsyncServiceInterface { HANDLER(KeepAlive); HANDLER(CloseContext); HANDLER(RegisterFunction); + HANDLER(SendTensor); #undef HANDLER + const WorkerEnv* const env_; // Not owned. EagerServiceImpl local_impl_; std::unique_ptr<::grpc::Alarm> shutdown_alarm_; @@ -71,8 +73,6 @@ class GrpcEagerServiceImpl : public AsyncServiceInterface { std::unique_ptr<::grpc::ServerCompletionQueue> cq_; tensorflow::eager::grpc::EagerService::AsyncService service_; - std::unique_ptr<thread::ThreadPool> request_handler_threadpool_; - TF_DISALLOW_COPY_AND_ASSIGN(GrpcEagerServiceImpl); }; |