diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/fft_ops_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/fft_ops_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/fft_ops_test.py b/tensorflow/python/kernel_tests/fft_ops_test.py index 629acedda5..f117934e4b 100644 --- a/tensorflow/python/kernel_tests/fft_ops_test.py +++ b/tensorflow/python/kernel_tests/fft_ops_test.py @@ -496,7 +496,7 @@ class RFFTOpsTest(BaseFFTOpsTest): "Input dimension .* must have length of at least 6 but got: 5"): x = np.zeros((5,) * rank).astype(np.float32) fft_length = [6] * rank - with self.test_session(): + with self.cached_session(): rfft_fn(x, fft_length).eval() with self.assertRaisesWithPredicateMatch( @@ -504,7 +504,7 @@ class RFFTOpsTest(BaseFFTOpsTest): "Input dimension .* must have length of at least .* but got: 3"): x = np.zeros((3,) * rank).astype(np.complex64) fft_length = [6] * rank - with self.test_session(): + with self.cached_session(): irfft_fn(x, fft_length).eval() def testGrad_Simple(self): |