Searched defs:forward_function (Results 1 – 5 of 5) sorted by relevance
182 forward_function=None): argument
2318 const tensorflow::eager::ForwardFunction<PyObject>* forward_function,2519 const tensorflow::eager::ForwardFunction<PyObject>* forward_function) {2550 PyObject* forward_function) {3211 tensorflow::eager::ForwardFunction<PyObject>* forward_function; local
994 def _wrap_forward_function_with_jvps( argument
1027 const ForwardFunction<Gradient>* forward_function, in Accumulate()
907 const py::handle& forward_function) { in PYBIND11_MODULE()