diff options
Diffstat (limited to 'tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable_test.py')
| -rw-r--r-- | tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable_test.py b/tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable_test.py index a2d82cf800..553b116a3b 100644 --- a/tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable_test.py +++ b/tensorflow/contrib/linear_optimizer/python/ops/sharded_mutable_dense_hashtable_test.py @@ -30,7 +30,7 @@ class ShardedMutableDenseHashTableTest(TensorFlowTestCase): def testShardedMutableHashTable(self): for num_shards in [1, 3, 10]: - with self.test_session(): + with self.cached_session(): default_val = -1 empty_key = 0 keys = constant_op.constant([11, 12, 13], dtypes.int64) @@ -53,7 +53,7 @@ class ShardedMutableDenseHashTableTest(TensorFlowTestCase): def testShardedMutableHashTableVectors(self): for num_shards in [1, 3, 10]: - with self.test_session(): + with self.cached_session(): default_val = [-0.1, 0.2] empty_key = [0, 1] keys = constant_op.constant([[11, 12], [13, 14], [15, 16]], @@ -79,7 +79,7 @@ class ShardedMutableDenseHashTableTest(TensorFlowTestCase): output.eval()) def testExportSharded(self): - with self.test_session(): + with self.cached_session(): empty_key = -2 default_val = -1 num_shards = 2 |