diff options
Diffstat (limited to 'tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.h')
| -rw-r--r-- | tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.h | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.h b/tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.h index 131de2863f..412395c526 100644 --- a/tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.h +++ b/tensorflow/core/distributed_runtime/rpc/grpc_master_service_impl.h @@ -25,7 +25,6 @@ limitations under the License. #include "grpc++/impl/codegen/stub_options.h" #include "grpc++/impl/codegen/sync_stream.h" -#include "tensorflow/core/distributed_runtime/rpc/grpc_namespace_compat.h" #include "tensorflow/core/distributed_runtime/rpc/grpc_serialization_traits.h" #include "tensorflow/core/protobuf/master.pb.h" @@ -108,13 +107,13 @@ class MasterService final { private: std::shared_ptr< ::grpc::ChannelInterface> channel_; - const ::grpc::RpcMethod rpcmethod_CreateSession_; - const ::grpc::RpcMethod rpcmethod_ExtendSession_; - const ::grpc::RpcMethod rpcmethod_PartialRunSetup_; - const ::grpc::RpcMethod rpcmethod_RunStep_; - const ::grpc::RpcMethod rpcmethod_CloseSession_; - const ::grpc::RpcMethod rpcmethod_ListDevices_; - const ::grpc::RpcMethod rpcmethod_Reset_; + const ::grpc::internal::RpcMethod rpcmethod_CreateSession_; + const ::grpc::internal::RpcMethod rpcmethod_ExtendSession_; + const ::grpc::internal::RpcMethod rpcmethod_PartialRunSetup_; + const ::grpc::internal::RpcMethod rpcmethod_RunStep_; + const ::grpc::internal::RpcMethod rpcmethod_CloseSession_; + const ::grpc::internal::RpcMethod rpcmethod_ListDevices_; + const ::grpc::internal::RpcMethod rpcmethod_Reset_; }; static std::unique_ptr<Stub> NewStub( const std::shared_ptr< ::grpc::ChannelInterface>& channel, |