diff options
Diffstat (limited to 'tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.cc')
-rw-r--r-- | tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.cc | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.cc b/tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.cc index 52e06c263d..f511674e1f 100644 --- a/tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.cc +++ b/tensorflow/core/distributed_runtime/rpc/eager/grpc_eager_service_impl.cc @@ -27,9 +27,7 @@ namespace eager { GrpcEagerServiceImpl::GrpcEagerServiceImpl( const WorkerEnv* env, ::grpc::ServerBuilder* server_builder) - : local_impl_(env) { - request_handler_threadpool_ = - MakeUnique<thread::ThreadPool>(env->env, "EagerServiceRequestHandler", 4); + : env_(env), local_impl_(env) { server_builder->RegisterService(&service_); cq_ = server_builder->AddCompletionQueue(); } @@ -50,6 +48,7 @@ void GrpcEagerServiceImpl::HandleRPCsLoop() { ENQUEUE_REQUEST(KeepAlive); ENQUEUE_REQUEST(CloseContext); ENQUEUE_REQUEST(RegisterFunction); + ENQUEUE_REQUEST(SendTensor); #undef ENQUEUE_REQUEST void* tag; // Matches the operation started against this cq_. |