diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-12 16:35:07 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-12 16:42:21 -0700 |
commit | 8f8b2497dbccf4b33557088b82b562205aa47c36 (patch) | |
tree | a26117486ba3d87a32c7abfd2c192f6f7501296c /tensorflow/contrib/solvers | |
parent | acc32e741935545d8e600a67361c388d14556538 (diff) |
Move from deprecated self.test_session() to self.cached_session().
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 212725342
Diffstat (limited to 'tensorflow/contrib/solvers')
4 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/contrib/solvers/python/kernel_tests/lanczos_test.py b/tensorflow/contrib/solvers/python/kernel_tests/lanczos_test.py index 4707dc2229..8fcd7aeef6 100644 --- a/tensorflow/contrib/solvers/python/kernel_tests/lanczos_test.py +++ b/tensorflow/contrib/solvers/python/kernel_tests/lanczos_test.py @@ -47,7 +47,7 @@ def _get_lanczos_tests(dtype_, use_static_shape_, shape_, orthogonalize_, low=-1.0, high=1.0, size=np.prod(shape_)).reshape(shape_).astype(dtype_) tol = 1e-12 if dtype_ == np.float64 else 1e-5 - with self.test_session() as sess: + with self.cached_session() as sess: if use_static_shape_: a = constant_op.constant(a_np) else: diff --git a/tensorflow/contrib/solvers/python/kernel_tests/least_squares_test.py b/tensorflow/contrib/solvers/python/kernel_tests/least_squares_test.py index a73642716b..2a9100903a 100644 --- a/tensorflow/contrib/solvers/python/kernel_tests/least_squares_test.py +++ b/tensorflow/contrib/solvers/python/kernel_tests/least_squares_test.py @@ -47,7 +47,7 @@ def _get_least_squares_tests(dtype_, use_static_shape_, shape_): low=-1.0, high=1.0, size=shape_[0]).astype(dtype_) tol = 1e-12 if dtype_ == np.float64 else 1e-6 max_iter = 20 - with self.test_session() as sess: + with self.cached_session() as sess: if use_static_shape_: a = constant_op.constant(a_np) rhs = constant_op.constant(rhs_np) diff --git a/tensorflow/contrib/solvers/python/kernel_tests/linear_equations_test.py b/tensorflow/contrib/solvers/python/kernel_tests/linear_equations_test.py index a1282847be..a0e6eb87bc 100644 --- a/tensorflow/contrib/solvers/python/kernel_tests/linear_equations_test.py +++ b/tensorflow/contrib/solvers/python/kernel_tests/linear_equations_test.py @@ -54,7 +54,7 @@ def _get_linear_equations_tests(dtype_, use_static_shape_, shape_): x_np = np.zeros_like(rhs_np) tol = 1e-6 if dtype_ == np.float64 else 1e-3 max_iter = 20 - with self.test_session() as sess: + with self.cached_session() as sess: if use_static_shape_: a = constant_op.constant(a_np) rhs = constant_op.constant(rhs_np) diff --git a/tensorflow/contrib/solvers/python/kernel_tests/util_test.py b/tensorflow/contrib/solvers/python/kernel_tests/util_test.py index 5d7534657b..57b4996689 100644 --- a/tensorflow/contrib/solvers/python/kernel_tests/util_test.py +++ b/tensorflow/contrib/solvers/python/kernel_tests/util_test.py @@ -33,7 +33,7 @@ class UtilTest(test.TestCase): a_np = np.array([[1., 2.], [3., 4.], [5., 6.]], dtype=dtype) x_np = np.array([[2.], [-3.]], dtype=dtype) y_np = np.array([[2], [-3.], [5.]], dtype=dtype) - with self.test_session() as sess: + with self.cached_session() as sess: if use_static_shape_: a = constant_op.constant(a_np, dtype=dtype) x = constant_op.constant(x_np, dtype=dtype) @@ -68,7 +68,7 @@ class UtilTest(test.TestCase): a_np = np.array([[1., 2.], [3., 4.], [5., 6.]], dtype=dtype) x_np = np.array([[2.], [-3.]], dtype=dtype) y_np = np.array([[2], [-3.], [5.]], dtype=dtype) - with self.test_session() as sess: + with self.cached_session() as sess: if use_static_shape_: a = constant_op.constant(a_np, dtype=dtype) x = constant_op.constant(x_np, dtype=dtype) @@ -101,7 +101,7 @@ class UtilTest(test.TestCase): self._testIdentityOperator(False) def testL2Norm(self): - with self.test_session(): + with self.cached_session(): x_np = np.array([[2], [-3.], [5.]]) x_norm_np = np.linalg.norm(x_np) x_normalized_np = x_np / x_norm_np |