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 …]
|