Searched refs:PyBackwardFunction (Results 1 – 1 of 1) sorted by relevance
54 typedef std::function<PyObject*(PyObject*)> PyBackwardFunction; typedef972 class PyVSpace : public tensorflow::eager::VSpace<PyObject, PyBackwardFunction,1093 PyBackwardFunction* backward_function, in CallBackwardFunction()1178 : public tensorflow::eager::GradientTape<PyObject, PyBackwardFunction,1182 : tensorflow::eager::GradientTape<PyObject, PyBackwardFunction, in GradientTape()1579 const std::function<PyBackwardFunction*()>& backward_function_getter,1580 const std::function<void(PyBackwardFunction*)>& backward_function_killer) {1638 PyBackwardFunction* function =1639 new PyBackwardFunction([backward_function](PyObject* out_grads) {1644 [backward_function](PyBackwardFunction* py_backward_function) {[all …]