diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/variable_ops_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/variable_ops_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/variable_ops_test.py b/tensorflow/python/kernel_tests/variable_ops_test.py index cf369c0718..3d2f8b6155 100644 --- a/tensorflow/python/kernel_tests/variable_ops_test.py +++ b/tensorflow/python/kernel_tests/variable_ops_test.py @@ -118,7 +118,7 @@ class VariableOpTest(test.TestCase): self.assertEqual(tensor_shape.unknown_shape(), assigned.get_shape()) def testAssignNoShape(self): - with self.test_session(): + with self.cached_session(): value = self._NewShapelessTensor() var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False) self.assertEqual(tensor_shape.unknown_shape(), var.get_shape()) @@ -126,7 +126,7 @@ class VariableOpTest(test.TestCase): state_ops.assign(var, value).get_shape()) def testAssignNoShapeNoValidateShape(self): - with self.test_session(): + with self.cached_session(): value = self._NewShapelessTensor() var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False) self.assertEqual(tensor_shape.unknown_shape(), var.get_shape()) |