diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/array_ops_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/array_ops_test.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/tensorflow/python/kernel_tests/array_ops_test.py b/tensorflow/python/kernel_tests/array_ops_test.py index 81442d12e9..b0e24e969c 100644 --- a/tensorflow/python/kernel_tests/array_ops_test.py +++ b/tensorflow/python/kernel_tests/array_ops_test.py @@ -559,6 +559,14 @@ class StridedSliceTest(test_util.TensorFlowTestCase): s = array_ops.strided_slice(x, begin, end, strides) self.assertAllEqual([3.], self.evaluate(s)) + @test_util.assert_no_new_pyobjects_executing_eagerly + def testEagerMemory(self): + with context.eager_mode(): + inputs = constant_op.constant( + [[[1], [2], [3], [4]]], dtype=dtypes.float32) + # Tests that slicing an EagerTensor doesn't leak memory + inputs[0] # pylint: disable=pointless-statement + def testDegenerateSlices(self): with self.test_session(use_gpu=True): checker = StridedSliceChecker(self, StridedSliceChecker.REF_TENSOR) @@ -1145,7 +1153,7 @@ class IdentityTest(test_util.TensorFlowTestCase): def testEagerIdentity(self): with context.eager_mode(): - ctx = context.get_default_context() + ctx = context.context() if not ctx.num_gpus(): self.skipTest("No GPUs found") |