diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/ops/seq2seq_ops_test.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/ops/seq2seq_ops_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/learn/python/learn/ops/seq2seq_ops_test.py b/tensorflow/contrib/learn/python/learn/ops/seq2seq_ops_test.py index 95aec61955..5a7e4ebfea 100644 --- a/tensorflow/contrib/learn/python/learn/ops/seq2seq_ops_test.py +++ b/tensorflow/contrib/learn/python/learn/ops/seq2seq_ops_test.py @@ -31,7 +31,7 @@ class Seq2SeqOpsTest(test.TestCase): """Sequence-to-sequence tests.""" def test_sequence_classifier(self): - with self.test_session() as session: + with self.cached_session() as session: decoding = [ array_ops.placeholder(dtypes.float32, [2, 2]) for _ in range(3) ] @@ -60,7 +60,7 @@ class Seq2SeqOpsTest(test.TestCase): def test_seq2seq_inputs(self): inp = np.array([[[1, 0], [0, 1], [1, 0]], [[0, 1], [1, 0], [0, 1]]]) out = np.array([[[0, 1, 0], [1, 0, 0]], [[1, 0, 0], [0, 1, 0]]]) - with self.test_session() as session: + with self.cached_session() as session: x = array_ops.placeholder(dtypes.float32, [2, 3, 2]) y = array_ops.placeholder(dtypes.float32, [2, 2, 3]) in_x, in_y, out_y = ops.seq2seq_inputs(x, y, 3, 2) @@ -77,7 +77,7 @@ class Seq2SeqOpsTest(test.TestCase): [[0, 0, 0], [0, 0, 0]]]) def test_rnn_decoder(self): - with self.test_session(): + with self.cached_session(): decoder_inputs = [ array_ops.placeholder(dtypes.float32, [2, 2]) for _ in range(3) ] |