diff options
-rw-r--r-- | tensorflow/contrib/BUILD | 2 | ||||
-rw-r--r-- | tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index 23bb783e22..ea17b928f2 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -143,6 +143,7 @@ cc_library( name = "contrib_kernels", visibility = ["//visibility:public"], deps = [ + "//tensorflow/contrib/bigtable:bigtable_kernels", "//tensorflow/contrib/boosted_trees:boosted_trees_kernels", "//tensorflow/contrib/coder:all_kernels", "//tensorflow/contrib/data/kernels:dataset_kernels", @@ -178,6 +179,7 @@ cc_library( name = "contrib_ops_op_lib", visibility = ["//visibility:public"], deps = [ + "//tensorflow/contrib/bigtable:bigtable_ops_op_lib", "//tensorflow/contrib/boosted_trees:boosted_trees_ops_op_lib", "//tensorflow/contrib/coder:all_ops", "//tensorflow/contrib/data:dataset_ops_op_lib", diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc b/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc index a6755a3496..1790b4bc11 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc +++ b/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc @@ -84,6 +84,8 @@ class BigtableClientOp : public OpKernel { channel_args.SetMaxReceiveMessageSize( max_receive_message_size_); channel_args.SetUserAgentPrefix("tensorflow"); + channel_args.SetInt(GRPC_ARG_KEEPALIVE_PERMIT_WITHOUT_CALLS, 0); + channel_args.SetInt(GRPC_ARG_KEEPALIVE_TIMEOUT_MS, 60 * 1000); client_options.set_channel_arguments(channel_args); std::shared_ptr<google::cloud::bigtable::DataClient> client = google::cloud::bigtable::CreateDefaultDataClient( |