diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/dense_update_ops_no_tsan_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/dense_update_ops_no_tsan_test.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/python/kernel_tests/dense_update_ops_no_tsan_test.py b/tensorflow/python/kernel_tests/dense_update_ops_no_tsan_test.py index d33bf1ba12..affbaf159d 100644 --- a/tensorflow/python/kernel_tests/dense_update_ops_no_tsan_test.py +++ b/tensorflow/python/kernel_tests/dense_update_ops_no_tsan_test.py @@ -33,7 +33,7 @@ class AssignOpTest(test.TestCase): # contain benign and deliberate data races when multiple threads update # the same parameters without a lock. def testParallelUpdateWithoutLocking(self): - with self.test_session() as sess: + with self.cached_session() as sess: ones_t = array_ops.fill([1024, 1024], 1.0) p = variables.Variable(array_ops.zeros([1024, 1024])) adds = [ @@ -60,7 +60,7 @@ class AssignOpTest(test.TestCase): self.assertTrue((vals <= ones * 20).all()) def testParallelAssignWithoutLocking(self): - with self.test_session() as sess: + with self.cached_session() as sess: ones_t = array_ops.fill([1024, 1024], float(1)) p = variables.Variable(array_ops.zeros([1024, 1024])) assigns = [ @@ -92,7 +92,7 @@ class AssignOpTest(test.TestCase): # returning the output tensors. This issue will be resolved with the new # resource variables. def testParallelUpdateWithLocking(self): - with self.test_session() as sess: + with self.cached_session() as sess: zeros_t = array_ops.fill([1024, 1024], 0.0) ones_t = array_ops.fill([1024, 1024], 1.0) p = variables.Variable(zeros_t) @@ -119,7 +119,7 @@ class AssignOpTest(test.TestCase): self.assertAllEqual(vals, ones * 20) def testParallelAssignWithLocking(self): - with self.test_session() as sess: + with self.cached_session() as sess: zeros_t = array_ops.fill([1024, 1024], 0.0) ones_t = array_ops.fill([1024, 1024], 1.0) p = variables.Variable(zeros_t) |