Exemple #1
0
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
Exemple #2
0
 def watch(self, tensor):
   """Adds a tensor to the tape."""
   pywrap_tensorflow.TFE_Py_TapeWatch(self._tape, tid(tensor))