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.py173 with tape.VariableWatcher() as variable_watcher:
177 self.assertAllEqual(variable_watcher.watched_variables(), (var1, var2))
182 with tape.VariableWatcher() as variable_watcher:
186 self.assertAllEqual(variable_watcher.watched_variables(), (var1,))
202 with tape.VariableWatcher() as variable_watcher:
208 self.assertAllEqual(variable_watcher.watched_variables(), (var1, var2))
Dpywrap_tfe.h347 void TFE_Py_VariableWatcherRemove(PyObject* variable_watcher);
354 PyObject* TFE_Py_VariableWatcherWatchedVariables(PyObject* variable_watcher);
Dpywrap_tfe_src.cc1701 VariableWatcher* variable_watcher; member
1704 static void TFE_Py_VariableWatcher_Delete(PyObject* variable_watcher) { in TFE_Py_VariableWatcher_Delete() argument
1705 delete reinterpret_cast<TFE_Py_VariableWatcher*>(variable_watcher) in TFE_Py_VariableWatcher_Delete()
1706 ->variable_watcher; in TFE_Py_VariableWatcher_Delete()
1707 Py_TYPE(variable_watcher)->tp_free(variable_watcher); in TFE_Py_VariableWatcher_Delete()
2290 TFE_Py_VariableWatcher* variable_watcher = local
2292 variable_watcher->variable_watcher = new VariableWatcher();
2293 Py_INCREF(variable_watcher);
2294 GetVariableWatcherSet()->insert(variable_watcher);
2295 return reinterpret_cast<PyObject*>(variable_watcher);
[all …]
/external/tensorflow/tensorflow/python/ops/
Dcustom_gradient.py420 with tape_lib.VariableWatcher() as variable_watcher:
443 v.ref() for v in variable_watcher.watched_variables()
530 with tape_lib.VariableWatcher() as variable_watcher:
538 for v in set(v.ref() for v in variable_watcher.watched_variables())
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc1120 m.def("TFE_Py_VariableWatcherRemove", [](const py::handle& variable_watcher) { in PYBIND11_MODULE() argument
1121 TFE_Py_VariableWatcherRemove(variable_watcher.ptr()); in PYBIND11_MODULE()
1128 [](const py::handle& variable_watcher) { in PYBIND11_MODULE() argument
1130 TFE_Py_VariableWatcherWatchedVariables(variable_watcher.ptr())); in PYBIND11_MODULE()