diff options
Diffstat (limited to 'tensorflow/core/distributed_runtime/rpc/grpc_worker_service_impl.cc')
-rw-r--r-- | tensorflow/core/distributed_runtime/rpc/grpc_worker_service_impl.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc/grpc_worker_service_impl.cc b/tensorflow/core/distributed_runtime/rpc/grpc_worker_service_impl.cc index 80a2f89337..348c6dc98b 100644 --- a/tensorflow/core/distributed_runtime/rpc/grpc_worker_service_impl.cc +++ b/tensorflow/core/distributed_runtime/rpc/grpc_worker_service_impl.cc @@ -58,9 +58,9 @@ namespace grpc { WorkerService::AsyncService::AsyncService() { for (int i = 0; i < kGrpcNumWorkerMethods; ++i) { - AddMethod(new ::grpc::RpcServiceMethod( + AddMethod(new ::grpc::internal::RpcServiceMethod( GrpcWorkerMethodName(static_cast<GrpcWorkerMethod>(i)), - ::grpc::RpcMethod::NORMAL_RPC, nullptr)); + ::grpc::internal::RpcMethod::NORMAL_RPC, nullptr)); ::grpc::Service::MarkMethodAsync(i); } } |