def __del__(self): # Make sure the resource is deleted in the same mode as it was created in. if self._eager_mode: with context.eager_mode(): gen_dataset_ops.delete_iterator(handle=self._handle, deleter=self._deleter) else: with context.graph_mode(): gen_dataset_ops.delete_iterator(handle=self._handle, deleter=self._deleter)
def __del__(self): with ops.device(self._device): # Make sure the resource is deleted in the same mode as it was created in. if self._eager_mode: with context.eager_mode(): gen_dataset_ops.delete_iterator( handle=self._handle, deleter=self._deleter) else: with context.graph_mode(): gen_dataset_ops.delete_iterator( handle=self._handle, deleter=self._deleter)