Home
last modified time | relevance | path

Searched refs:variable_watcher (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dtape_test.py176 with tape.VariableWatcher() as variable_watcher:
180 self.assertAllEqual(variable_watcher.watched_variables(), (var1, var2))
185 with tape.VariableWatcher() as variable_watcher:
189 self.assertAllEqual(variable_watcher.watched_variables(), (var1,))
205 with tape.VariableWatcher() as variable_watcher:
211 self.assertAllEqual(variable_watcher.watched_variables(), (var1, var2))
Dpywrap_tfe.h343 void TFE_Py_VariableWatcherRemove(PyObject* variable_watcher);
350 PyObject* TFE_Py_VariableWatcherWatchedVariables(PyObject* variable_watcher);
Dpywrap_tfe_src.cc1656 VariableWatcher* variable_watcher; member
1659 static void TFE_Py_VariableWatcher_Delete(PyObject* variable_watcher) { in TFE_Py_VariableWatcher_Delete() argument
1660 delete reinterpret_cast<TFE_Py_VariableWatcher*>(variable_watcher) in TFE_Py_VariableWatcher_Delete()
1661 ->variable_watcher; in TFE_Py_VariableWatcher_Delete()
1662 Py_TYPE(variable_watcher)->tp_free(variable_watcher); in TFE_Py_VariableWatcher_Delete()
2220 TFE_Py_VariableWatcher* variable_watcher = local
2222 variable_watcher->variable_watcher = new VariableWatcher();
2223 Py_INCREF(variable_watcher);
2224 GetVariableWatcherSet()->insert(variable_watcher);
2225 return reinterpret_cast<PyObject*>(variable_watcher);
[all …]
/external/tensorflow/tensorflow/python/ops/
Dcustom_gradient.py329 with tape_lib.VariableWatcher() as variable_watcher:
352 v.ref() for v in variable_watcher.watched_variables()
437 with tape_lib.VariableWatcher() as variable_watcher:
445 for v in set(v.ref() for v in variable_watcher.watched_variables())
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc961 m.def("TFE_Py_VariableWatcherRemove", [](const py::handle& variable_watcher) { in PYBIND11_MODULE() argument
962 TFE_Py_VariableWatcherRemove(variable_watcher.ptr()); in PYBIND11_MODULE()
969 [](const py::handle& variable_watcher) { in PYBIND11_MODULE() argument
971 TFE_Py_VariableWatcherWatchedVariables(variable_watcher.ptr())); in PYBIND11_MODULE()