aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.cc')
-rw-r--r--tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.cc35
1 files changed, 18 insertions, 17 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.cc b/tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.cc
index c42622dd50..17d0047eb2 100644
--- a/tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.cc
+++ b/tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.cc
@@ -49,74 +49,75 @@ MasterService::Stub::Stub(
const std::shared_ptr< ::grpc::ChannelInterface>& channel)
: channel_(channel),
rpcmethod_CreateSession_(grpcMasterService_method_names[0],
- ::grpc::RpcMethod::NORMAL_RPC, channel),
+ ::grpc::internal::RpcMethod::NORMAL_RPC, channel),
rpcmethod_ExtendSession_(grpcMasterService_method_names[1],
- ::grpc::RpcMethod::NORMAL_RPC, channel),
+ ::grpc::internal::RpcMethod::NORMAL_RPC, channel),
rpcmethod_PartialRunSetup_(grpcMasterService_method_names[2],
- ::grpc::RpcMethod::NORMAL_RPC, channel),
+ ::grpc::internal::RpcMethod::NORMAL_RPC, channel),
rpcmethod_RunStep_(grpcMasterService_method_names[3],
- ::grpc::RpcMethod::NORMAL_RPC, channel),
+ ::grpc::internal::RpcMethod::NORMAL_RPC, channel),
rpcmethod_CloseSession_(grpcMasterService_method_names[4],
- ::grpc::RpcMethod::NORMAL_RPC, channel),
+ ::grpc::internal::RpcMethod::NORMAL_RPC, channel),
rpcmethod_ListDevices_(grpcMasterService_method_names[5],
- ::grpc::RpcMethod::NORMAL_RPC, channel),
+ ::grpc::internal::RpcMethod::NORMAL_RPC, channel),
rpcmethod_Reset_(grpcMasterService_method_names[6],
- ::grpc::RpcMethod::NORMAL_RPC, channel) {}
+ ::grpc::internal::RpcMethod::NORMAL_RPC, channel) {}
::grpc::Status MasterService::Stub::CreateSession(
::grpc::ClientContext* context, const CreateSessionRequest& request,
CreateSessionResponse* response) {
- return ::grpc::BlockingUnaryCall(channel_.get(), rpcmethod_CreateSession_,
+ return ::grpc::internal::BlockingUnaryCall(channel_.get(), rpcmethod_CreateSession_,
context, request, response);
}
::grpc::Status MasterService::Stub::ExtendSession(
::grpc::ClientContext* context, const ExtendSessionRequest& request,
ExtendSessionResponse* response) {
- return ::grpc::BlockingUnaryCall(channel_.get(), rpcmethod_ExtendSession_,
+ return ::grpc::internal::BlockingUnaryCall(channel_.get(), rpcmethod_ExtendSession_,
context, request, response);
}
::grpc::Status MasterService::Stub::PartialRunSetup(
::grpc::ClientContext* context, const PartialRunSetupRequest& request,
PartialRunSetupResponse* response) {
- return ::grpc::BlockingUnaryCall(channel_.get(), rpcmethod_PartialRunSetup_,
+ return ::grpc::internal::BlockingUnaryCall(channel_.get(), rpcmethod_PartialRunSetup_,
context, request, response);
}
::grpc::Status MasterService::Stub::RunStep(::grpc::ClientContext* context,
const RunStepRequest& request,
RunStepResponse* response) {
- return ::grpc::BlockingUnaryCall(channel_.get(), rpcmethod_RunStep_, context,
+ return ::grpc::internal::BlockingUnaryCall(channel_.get(), rpcmethod_RunStep_, context,
request, response);
}
::grpc::Status MasterService::Stub::CloseSession(
::grpc::ClientContext* context, const CloseSessionRequest& request,
CloseSessionResponse* response) {
- return ::grpc::BlockingUnaryCall(channel_.get(), rpcmethod_CloseSession_,
+ return ::grpc::internal::BlockingUnaryCall(channel_.get(), rpcmethod_CloseSession_,
context, request, response);
}
::grpc::Status MasterService::Stub::ListDevices(
::grpc::ClientContext* context, const ListDevicesRequest& request,
ListDevicesResponse* response) {
- return ::grpc::BlockingUnaryCall(channel_.get(), rpcmethod_ListDevices_,
+ return ::grpc::internal::BlockingUnaryCall(channel_.get(), rpcmethod_ListDevices_,
context, request, response);
}
::grpc::Status MasterService::Stub::Reset(::grpc::ClientContext* context,
const ResetRequest& request,
ResetResponse* response) {
- return ::grpc::BlockingUnaryCall(channel_.get(), rpcmethod_Reset_, context,
+ return ::grpc::internal::BlockingUnaryCall(channel_.get(), rpcmethod_Reset_, context,
request, response);
}
MasterService::AsyncService::AsyncService() {
for (int i = 0; i < 7; ++i) {
- AddMethod(new ::grpc::RpcServiceMethod(grpcMasterService_method_names[i],
- ::grpc::RpcMethod::NORMAL_RPC,
- nullptr));
+ AddMethod(new ::grpc::internal::RpcServiceMethod(
+ grpcMasterService_method_names[i],
+ ::grpc::internal::RpcMethod::NORMAL_RPC,
+ nullptr));
::grpc::Service::MarkMethodAsync(i);
}
}