OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 12:20:24 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Always run iterator deleter in eager mode for safety.

Browse Source 
PiperOrigin-RevId: 173947019
This commit is contained in:
Alexandre Passos 2017-10-30 14:10:55 -07:00 committed by TensorFlower Gardener
parent efcbf6e34e
commit 3639aa7ff1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/contrib/eager/python/datasets.py
View File

@ -84,7 +84,7 @@ class Iterator(object):
def __del__(self):
if self._resource is not None:
with ops.device("/device:CPU:0"):
with ops.device("/device:CPU:0"), context.eager_mode():
resource_variable_ops.destroy_resource_op(self._resource)
self._resource = None