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/core/framework/lookup_interface.cc | |
parent | 2f5ebc0ea5e6d500ea8cd925234c569d6b32fd4e (diff) |
Add 'remove' operation to MutableHashTable and MutableDenseHashTable.
PiperOrigin-RevId: 216443201
Diffstat (limited to 'tensorflow/core/framework/lookup_interface.cc')
-rw-r--r-- | tensorflow/core/framework/lookup_interface.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/core/framework/lookup_interface.cc b/tensorflow/core/framework/lookup_interface.cc index bf3204ea6e..117adbf65c 100644 --- a/tensorflow/core/framework/lookup_interface.cc +++ b/tensorflow/core/framework/lookup_interface.cc @@ -71,6 +71,14 @@ Status LookupInterface::CheckKeyAndValueTensorsForImport(const Tensor& keys, return CheckKeyAndValueTensorsHelper(keys, values); } +Status LookupInterface::CheckKeyTensorForRemove(const Tensor& keys) { + if (keys.dtype() != key_dtype()) { + return errors::InvalidArgument("Key must be type ", key_dtype(), + " but got ", keys.dtype()); + } + return CheckKeyShape(keys.shape()); +} + Status LookupInterface::CheckFindArguments(const Tensor& key, const Tensor& default_value) { TF_RETURN_IF_ERROR(CheckKeyAndValueTypes(key, default_value)); |