Searched refs:PyBackwardFunction (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | pywrap_tfe_src.cc | 128 PyBackwardFunction; typedef
1229 class PyVSpace : public tensorflow::eager::VSpace<PyObject, PyBackwardFunction,
1370 const string& op_type, PyBackwardFunction* backward_function, in CallBackwardFunction()
1573 : public tensorflow::eager::GradientTape<PyObject, PyBackwardFunction,
1577 : tensorflow::eager::GradientTape<PyObject, PyBackwardFunction, in GradientTape()
1606 typedef tensorflow::eager::ForwardAccumulator<PyObject, PyBackwardFunction,
2368 const std::function<PyBackwardFunction*()>& backward_function_getter,
2369 const std::function<void(PyBackwardFunction*)>& backward_function_killer,
2388 const std::function<PyBackwardFunction*()>& backward_function_getter,
2389 const std::function<void(PyBackwardFunction*)>& backward_function_killer,
[all …]
|