diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-09 18:45:22 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-09 18:53:14 -0700 |
| commit | 79387568f2860dd25f411a2e3ba764dabb76286d (patch) | |
| tree | 2f5dba53ebc10ab90f9f6cd86a5e7fed9f1a2601 | |
| parent | 367903193f44b16e79981f2f29f4f2ceea8b6117 (diff) | |
Permit TensorFlow server to access Cloud Bigtable.
PiperOrigin-RevId: 208146858
| -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( |