diff options
author | Allen Lavoie <allenl@google.com> | 2018-03-09 14:42:51 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-09 14:52:20 -0800 |
commit | 88c75b081fe17f04c58c954a76d673abd15255cb (patch) | |
tree | 6e1de36b6a9bfcee64fd2b2a8e140c95bc597830 /tensorflow/python/layers | |
parent | 41fe9b97b90dd67479ac57fda94ce5ee862df960 (diff) |
Eager: Fix a Dimension PyObject leak, test for it.
PiperOrigin-RevId: 188540944
Diffstat (limited to 'tensorflow/python/layers')
-rw-r--r-- | tensorflow/python/layers/core_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/layers/core_test.py b/tensorflow/python/layers/core_test.py index ae19866d7a..09287e4906 100644 --- a/tensorflow/python/layers/core_test.py +++ b/tensorflow/python/layers/core_test.py @@ -67,7 +67,7 @@ class DenseTest(test.TestCase): variables.global_variables_initializer().run() self.assertAllEqual(x.eval(), [[0.0]]) - @test_util.run_in_graph_and_eager_modes() + @test_util.run_in_graph_and_eager_modes(assert_no_eager_garbage=True) def testCall(self): dense = core_layers.Dense(2, activation=nn_ops.relu, name='my_dense') inputs = random_ops.random_uniform((5, 4), seed=1) |