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/timeseries/python/timeseries/math_utils_test.py | |
parent | 2f5ebc0ea5e6d500ea8cd925234c569d6b32fd4e (diff) |
Add 'remove' operation to MutableHashTable and MutableDenseHashTable.
PiperOrigin-RevId: 216443201
Diffstat (limited to 'tensorflow/contrib/timeseries/python/timeseries/math_utils_test.py')
-rw-r--r-- | tensorflow/contrib/timeseries/python/timeseries/math_utils_test.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tensorflow/contrib/timeseries/python/timeseries/math_utils_test.py b/tensorflow/contrib/timeseries/python/timeseries/math_utils_test.py index c0de42b15b..91265b9b2e 100644 --- a/tensorflow/contrib/timeseries/python/timeseries/math_utils_test.py +++ b/tensorflow/contrib/timeseries/python/timeseries/math_utils_test.py @@ -223,10 +223,12 @@ class TestLookupTable(test.TestCase): hash_table = math_utils.TupleOfTensorsLookup( key_dtype=dtypes.int64, default_values=[[ - array_ops.ones([3, 2], dtype=dtypes.float32), array_ops.zeros( - [5], dtype=dtypes.float64) - ], array_ops.ones([7, 7], dtype=dtypes.int64)], + array_ops.ones([3, 2], dtype=dtypes.float32), + array_ops.zeros([5], dtype=dtypes.float64) + ], + array_ops.ones([7, 7], dtype=dtypes.int64)], empty_key=-1, + deleted_key=-2, name="test_lookup") def stack_tensor(base_tensor): return array_ops.stack([base_tensor + 1, base_tensor + 2]) |