Searched defs:forward_function (Results 1 – 5 of 5) sorted by relevance
178 forward_function=None): argument
2390 const tensorflow::eager::ForwardFunction<PyObject>* forward_function,2591 const tensorflow::eager::ForwardFunction<PyObject>* forward_function) {2622 PyObject* forward_function) {3290 tensorflow::eager::ForwardFunction<PyObject>* forward_function; local
897 def _wrap_forward_function_with_jvps( argument
1027 const ForwardFunction<Gradient>* forward_function, in Accumulate()
1066 const py::handle& forward_function) { in PYBIND11_MODULE()