diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/cross_grad_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/cross_grad_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/kernel_tests/cross_grad_test.py b/tensorflow/python/kernel_tests/cross_grad_test.py index f040ac6055..0bd4006d6a 100644 --- a/tensorflow/python/kernel_tests/cross_grad_test.py +++ b/tensorflow/python/kernel_tests/cross_grad_test.py @@ -27,7 +27,7 @@ from tensorflow.python.platform import test class CrossOpTest(test.TestCase): def testGradientRandomValues(self): - with self.test_session(): + with self.cached_session(): us = [2, 3] u = array_ops.reshape( [0.854, -0.616, 0.767, 0.725, -0.927, 0.159], shape=us) |