diff options
Diffstat (limited to 'tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc')
-rw-r--r-- | tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc b/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc index 9b276ec676..70923e6287 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc +++ b/tensorflow/contrib/bigtable/kernels/bigtable_kernels.cc @@ -76,8 +76,10 @@ class BigtableClientOp : public OpKernel { cinfo_.container(), cinfo_.name(), &resource, [this, ctx]( BigtableClientResource** ret) EXCLUSIVE_LOCKS_REQUIRED(mu_) { - auto client_options = google::cloud::bigtable::ClientOptions(); - client_options.set_connection_pool_size(connection_pool_size_); + auto client_options = + google::cloud::bigtable::ClientOptions() + .set_connection_pool_size(connection_pool_size_) + .set_data_endpoint("batch-bigtable.googleapis.com"); auto channel_args = client_options.channel_arguments(); channel_args.SetMaxReceiveMessageSize( max_receive_message_size_); |