def watch(tape, tensor): """Marks this tensor to be watched by the given tape.""" pywrap_tensorflow.TFE_Py_TapeWatch(tape._tape, tensor) # pylint: disable=protected-access
def watch(self, tensor): """Adds a tensor to the tape.""" pywrap_tensorflow.TFE_Py_TapeWatch(self._tape, tid(tensor))