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 /tensorflow/contrib/bigtable | |
| parent | 367903193f44b16e79981f2f29f4f2ceea8b6117 (diff) | |
Permit TensorFlow server to access Cloud Bigtable.
PiperOrigin-RevId: 208146858
Diffstat (limited to 'tensorflow/contrib/bigtable')
| -rw-r--r-- | tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc | 2 |
1 files changed, 2 insertions, 0 deletions
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( |