aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/verbs
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/verbs')
-rw-r--r--tensorflow/contrib/verbs/grpc_verbs_service_impl.cc14
-rw-r--r--tensorflow/contrib/verbs/grpc_verbs_service_impl.h2
2 files changed, 9 insertions, 7 deletions
diff --git a/tensorflow/contrib/verbs/grpc_verbs_service_impl.cc b/tensorflow/contrib/verbs/grpc_verbs_service_impl.cc
index e0ba78dbfd..a1fbea57dd 100644
--- a/tensorflow/contrib/verbs/grpc_verbs_service_impl.cc
+++ b/tensorflow/contrib/verbs/grpc_verbs_service_impl.cc
@@ -43,20 +43,22 @@ VerbsService::Stub::Stub(
const std::shared_ptr< ::grpc::ChannelInterface>& channel)
: channel_(channel),
rpcmethod_GetRemoteAddress_(grpcVerbsService_method_names[0],
- ::grpc::RpcMethod::NORMAL_RPC, channel) {}
+ ::grpc::internal::RpcMethod::NORMAL_RPC,
+ channel) {}
::grpc::Status VerbsService::Stub::GetRemoteAddress(
::grpc::ClientContext* context, const GetRemoteAddressRequest& request,
GetRemoteAddressResponse* response) {
- return ::grpc::BlockingUnaryCall(channel_.get(), rpcmethod_GetRemoteAddress_,
- context, request, response);
+ return ::grpc::internal::BlockingUnaryCall(
+ channel_.get(), rpcmethod_GetRemoteAddress_, context, request, response);
}
VerbsService::AsyncService::AsyncService() {
for (int i = 0; i < 1; ++i) {
- AddMethod(new ::grpc::RpcServiceMethod(grpcVerbsService_method_names[i],
- ::grpc::RpcMethod::NORMAL_RPC,
- nullptr));
+ AddMethod(new ::grpc::internal::RpcServiceMethod(
+ grpcVerbsService_method_names[i],
+ ::grpc::internal::RpcMethod::NORMAL_RPC,
+ nullptr));
::grpc::Service::MarkMethodAsync(i);
}
}
diff --git a/tensorflow/contrib/verbs/grpc_verbs_service_impl.h b/tensorflow/contrib/verbs/grpc_verbs_service_impl.h
index f7ea774b66..86431ca030 100644
--- a/tensorflow/contrib/verbs/grpc_verbs_service_impl.h
+++ b/tensorflow/contrib/verbs/grpc_verbs_service_impl.h
@@ -61,7 +61,7 @@ class VerbsService GRPC_FINAL {
private:
std::shared_ptr< ::grpc::ChannelInterface> channel_;
- const ::grpc::RpcMethod rpcmethod_GetRemoteAddress_;
+ const ::grpc::internal::RpcMethod rpcmethod_GetRemoteAddress_;
};
static std::unique_ptr<Stub> NewStub(
const std::shared_ptr< ::grpc::ChannelInterface>& channel,