diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-10 13:57:11 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-10 14:06:20 -0700 |
| commit | f7cb9d504f1a675375171fa19cee70fa64c28c64 (patch) | |
| tree | df105421126ce8c3f0ef856acb61beef9851a3fc /tensorflow/contrib/bigtable | |
| parent | db294c86cc299d5fc8c5cbf089b07e59723a4664 (diff) | |
Permit TensorFlow server to access Cloud Bigtable.
PiperOrigin-RevId: 208263100
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( |