From 9713632f864c95f19537589e6b34e05749390d6b Mon Sep 17 00:00:00 2001 From: Ping Yu <4018+pyu10055@users.noreply.github.com> Date: Fri, 1 Nov 2019 09:10:59 -0700 Subject: [PATCH] sync g3 change to github --- .../converters/tf_saved_model_conversion_v2_test.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2_test.py b/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2_test.py index 0f44fca6053..8df5fde2ec8 100644 --- a/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2_test.py +++ b/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2_test.py @@ -23,6 +23,7 @@ import unittest import tensorflow as tf +from tensorflow.contrib import lookup as contrib_lookup from tensorflow.python.eager import def_function from tensorflow.python.framework import constant_op from tensorflow.python.framework import dtypes @@ -94,8 +95,8 @@ def _create_saved_model_v1_with_hashtable(self): # Add a hash table that is not used by the output. keys = tf.constant(['key']) values = tf.constant([1]) - initializer = tf.contrib.lookup.KeyValueTensorInitializer(keys, values) - table = tf.contrib.lookup.HashTable(initializer, -1) + initializer = contrib_lookup.KeyValueTensorInitializer(keys, values) + table = contrib_lookup.HashTable(initializer, -1) # Create a builder. save_dir = os.path.join(self._tmp_dir, SAVED_MODEL_DIR)