diff options
author | Brennan Saeta <saeta@google.com> | 2018-07-12 11:17:14 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-12 11:26:16 -0700 |
commit | 365d2fc4d62540b2c6524500a7a58e7edab0dfa9 (patch) | |
tree | 4632492ab41c0c1b58209b47fef1b817b4979682 /tensorflow/contrib/bigtable | |
parent | d22484aebe10b7e98ed445e86b5cc445fcc71b71 (diff) |
[tf.data / Bigtable]: Set AlwaysRetryMutationPolicy
Because the tf.data integration currently doesn't support setting client-side timestamps, using the AlwaysRetryMutationPolicy can make data loading more reliable. (This is safe-ish to do, because when reading TF always read Latest(1), so duplicate writes will not be visible to user programs.)
PiperOrigin-RevId: 204331133
Diffstat (limited to 'tensorflow/contrib/bigtable')
-rw-r--r-- | tensorflow/contrib/bigtable/kernels/bigtable_lib.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/bigtable/kernels/bigtable_lib.h b/tensorflow/contrib/bigtable/kernels/bigtable_lib.h index 12d8256dea..a2a5df1037 100644 --- a/tensorflow/contrib/bigtable/kernels/bigtable_lib.h +++ b/tensorflow/contrib/bigtable/kernels/bigtable_lib.h @@ -58,7 +58,8 @@ class BigtableTableResource : public ResourceBase { BigtableTableResource(BigtableClientResource* client, string table_name) : client_(client), table_name_(std::move(table_name)), - table_(client->get_client(), table_name_) { + table_(client->get_client(), table_name_, + google::cloud::bigtable::AlwaysRetryMutationPolicy()) { client_->Ref(); } |