diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-09 16:23:35 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 16:39:23 -0700 |
| commit | 6c391166b8b6ba43d2b0151e6fb9cf14864131a2 (patch) | |
| tree | c8c3c9eadade00f1a4e6cec2024e2a15bfd0b948 /tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable.py | |
| parent | 2f5ebc0ea5e6d500ea8cd925234c569d6b32fd4e (diff) | |
Add 'remove' operation to MutableHashTable and MutableDenseHashTable.
PiperOrigin-RevId: 216443201
Diffstat (limited to 'tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable.py')
| -rw-r--r-- | tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable.py b/tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable.py index 5015fb0848..44a869f7c2 100644 --- a/tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable.py +++ b/tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable.py @@ -48,6 +48,7 @@ class ShardedMutableDenseHashTable(lookup.LookupInterface): value_dtype, default_value, empty_key, + deleted_key, num_shards=1, checkpoint=True, name='ShardedMutableHashTable'): @@ -62,6 +63,7 @@ class ShardedMutableDenseHashTable(lookup.LookupInterface): value_dtype=value_dtype, default_value=default_value, empty_key=empty_key, + deleted_key=deleted_key, checkpoint=checkpoint, name='%s-%d-of-%d' % (name, i + 1, num_shards))) self._table_shards = table_shards |