diff --git a/tensorflow/core/debug/debug_io_utils.cc b/tensorflow/core/debug/debug_io_utils.cc index 6994dec3b59..527c9353801 100644 --- a/tensorflow/core/debug/debug_io_utils.cc +++ b/tensorflow/core/debug/debug_io_utils.cc @@ -730,7 +730,7 @@ Status DebugGrpcChannel::Connect(const int64 timeout_micros) { ::grpc::ChannelArguments args; args.SetInt(GRPC_ARG_MAX_MESSAGE_LENGTH, std::numeric_limits::max()); // Avoid problems where default reconnect backoff is too long (e.g., 20 s). - args.SetInt("grpc.testing.fixed_reconnect_backoff_ms", 1000); + args.SetInt(GRPC_ARG_MAX_RECONNECT_BACKOFF_MS, 1000); channel_ = ::grpc::CreateCustomChannel( server_stream_addr_, ::grpc::InsecureChannelCredentials(), args); if (!channel_->WaitForConnected( diff --git a/tensorflow/core/distributed_runtime/rpc/grpc_channel.cc b/tensorflow/core/distributed_runtime/rpc/grpc_channel.cc index 781b7d65cdd..64c221805b0 100644 --- a/tensorflow/core/distributed_runtime/rpc/grpc_channel.cc +++ b/tensorflow/core/distributed_runtime/rpc/grpc_channel.cc @@ -62,7 +62,7 @@ Status ValidateHostPortPair(const string& host_port) { args.SetInt(GRPC_ARG_MAX_MESSAGE_LENGTH, std::numeric_limits::max()); // NOTE(mrry): Some versions of gRPC use a 20-second minimum backoff // on connection failure, which makes our tests time out. - args.SetInt("grpc.testing.fixed_reconnect_backoff_ms", 1000); + args.SetInt(GRPC_ARG_MAX_RECONNECT_BACKOFF_MS, 1000); if (rpc_options != nullptr) { if (rpc_options->compression_algorithm() == "deflate") { args.SetCompressionAlgorithm(GRPC_COMPRESS_DEFLATE);