diff options
author | 2018-08-24 17:12:31 -0700 | |
---|---|---|
committer | 2018-08-24 17:16:16 -0700 | |
commit | 196c5b3450a31093eb5acc17df31300f1c3f56bd (patch) | |
tree | 1336bc202f6430d25b20cd3c2d3bf9a611f2dcaa /tensorflow/core/kernels/initializable_lookup_table.cc | |
parent | f1b0140b311e0b888bfe8373716f252dfb5e0439 (diff) |
Changing KeyValueTensorInitializer to be more general.
This is similar to the previous version but replaces the initialize_table_v2 call with gen_lookup_ops.lookup_table_import_v2. They are conceptually doing the same thing but initialize_table_v2 requires table to be a child of InitializableLookupTable whereas lookup_table_import_v2 does not. So the new intializer is more general than previous one.
PiperOrigin-RevId: 210182320
Diffstat (limited to 'tensorflow/core/kernels/initializable_lookup_table.cc')
-rw-r--r-- | tensorflow/core/kernels/initializable_lookup_table.cc | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/initializable_lookup_table.cc b/tensorflow/core/kernels/initializable_lookup_table.cc index 06d53eba30..fcf468f5a8 100644 --- a/tensorflow/core/kernels/initializable_lookup_table.cc +++ b/tensorflow/core/kernels/initializable_lookup_table.cc @@ -14,7 +14,6 @@ limitations under the License. ==============================================================================*/ #include "tensorflow/core/kernels/initializable_lookup_table.h" - #include "tensorflow/core/lib/core/errors.h" namespace tensorflow { @@ -32,6 +31,13 @@ Status InitializableLookupTable::Find(OpKernelContext* ctx, const Tensor& keys, return DoFind(keys, values, default_value); } +Status InitializableLookupTable::ImportValues(OpKernelContext* ctx, + const Tensor& keys, + const Tensor& values) { + lookup::KeyValueTensorIterator iter(&keys, &values); + return Initialize(iter); +} + Status InitializableLookupTable::Initialize(InitTableIterator& iter) { if (!iter.Valid()) { return iter.status(); |