Home
last modified time | relevance | path

Searched defs:backward_function (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dtape.py177 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
186 backward_function): argument
194 backward_function, argument
H A Dpywrap_tfe_src.cc1370 const string& op_type, PyBackwardFunction* backward_function, in CallBackwardFunction()
2621 PyObject* backward_function,
2676 PyObject* backward_function, PyObject* forwardprop_output_indices) {
2727 PyObject* backward_function) {
3355 forward_pass_name_scope](PyBackwardFunction* backward_function) {
H A Dfunction.py898 self, forward_function, backward_function, argument
991 def _wrap_backward_function_with_jvp_backprop( argument
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dtfe_wrapper.cc1074 const py::handle& input_tensors, const py::handle& backward_function) { in PYBIND11_MODULE()
1083 const py::handle& forwardprop_output_indices) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dtape.h46 BackwardFunction* backward_function; member