Home
last modified time | relevance | path

Searched refs:_tape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dtape.py41 self._tape = tape
44 return pywrap_tfe.TFE_Py_TapeWatchedVariables(self._tape)
55 pywrap_tfe.TFE_Py_TapeSetAdd(tape._tape) # pylint: disable=protected-access
60 pywrap_tfe.TFE_Py_TapeWatch(tape._tape, tensor) # pylint: disable=protected-access
104 pywrap_tfe.TFE_Py_TapeWatchVariable(tape._tape, var) # pylint: disable=protected-access
151 pywrap_tfe.TFE_Py_TapeSetRemove(tape._tape) # pylint: disable=protected-access
Dbackprop.py841 self._tape = None
862 if self._tape is None:
863 self._tape = tape.push_new_tape(
867 tape.push_tape(self._tape)
873 tape.pop_tape(self._tape)
909 tape.watch_variable(self._tape, t)
911 tape.watch(self._tape, t)
936 if self._tape is None:
978 self._tape = None
983 if self._tape is not None:
[all …]
Dimperative_grad.py72 tape._tape, # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/framework/experimental/
Dtape.py21 from tensorflow.python.framework.experimental import _tape
31 self._c_tape = _tape.Tape(persistent)
33 self._tape_context = _tape.TapeContext(
Dgradient_registry.py21 from tensorflow.python.framework.experimental import _tape
23 _GRADIENT_REGISTRY_GLOBAL = _tape.GradientRegistry()
DBUILD31 name = "_tape",
34 module_name = "_tape",
84 deps = [":_tape"],
112 ":_tape",
Dtape.cc47 PYBIND11_MODULE(_tape, m) { in PYBIND11_MODULE() argument
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scaling_gradient_tape.py163 if self._tape is None: # pylint: disable=access-member-before-definition
180 self._tape = None # free up resources if a persistent tape was not needed