diff options
author | 2018-07-07 00:36:37 -0700 | |
---|---|---|
committer | 2018-07-07 21:38:24 -0700 | |
commit | 6d5b8b7cae669372df3f756f827c40b08a0d14a9 (patch) | |
tree | 9776fc3818bdea2bb9a154594d56e7e5c092317a /tensorflow/contrib/bigtable/ops | |
parent | 82d122cd1efc905070ed86ed55951b7437209523 (diff) |
[tf.data / Bigtable] Add max_receive_message_size connection parameter.
When storing images in Cloud Bigtable, the resulting gRPC messages are often larger than the default receive message max size value. This change makes the maximum receive message sizes configurable, and sets a more reasonable default for general TensorFlow use.
PiperOrigin-RevId: 203569796
Diffstat (limited to 'tensorflow/contrib/bigtable/ops')
-rw-r--r-- | tensorflow/contrib/bigtable/ops/bigtable_ops.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/bigtable/ops/bigtable_ops.cc b/tensorflow/contrib/bigtable/ops/bigtable_ops.cc index 179963457c..36a392f2a4 100644 --- a/tensorflow/contrib/bigtable/ops/bigtable_ops.cc +++ b/tensorflow/contrib/bigtable/ops/bigtable_ops.cc @@ -23,6 +23,7 @@ REGISTER_OP("BigtableClient") .Attr("project_id: string") .Attr("instance_id: string") .Attr("connection_pool_size: int") + .Attr("max_receive_message_size: int = -1") .Attr("container: string = ''") .Attr("shared_name: string = ''") .Output("client: resource") |