Searched refs:backward_function (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 40 BackwardFunction* backward_function; member
96 BackwardFunction* backward_function,
120 pair.second.backward_function_deleter(pair.second.backward_function); in ~GradientTape()
269 op_it->second.backward_function_deleter(op_it->second.backward_function); in DeleteTrace()
376 op_pair.second.backward_function); in PrepareBackprop()
496 pair.second.backward_function_deleter(pair.second.backward_function); in ComputeGradient()
560 Status s = vspace.CallBackwardFunction(trace.backward_function, in ComputeGradient()
563 trace.backward_function_deleter(trace.backward_function); in ComputeGradient()
572 trace.backward_function_deleter(trace.backward_function); in ComputeGradient()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | tape.py | 132 def record_operation(op_type, output_tensors, input_tensors, backward_function): argument
135 op_type, output_tensors, input_tensors, backward_function)
|
| D | pywrap_tfe.h | 169 PyObject* backward_function);
|
| D | function.py | 758 backward_function = self._backward_graph_function._inference_function
761 backward_function.add_to_graph(g)
846 def backward_function(*args): function
853 args, backward_function)
887 def backward_function(*args): function
891 args, backward_function)
|
| D | pywrap_tfe_src.cc | 1093 PyBackwardFunction* backward_function, in CallBackwardFunction() argument
1106 PyObject* py_result = (*backward_function)(grads); in CallBackwardFunction()
1623 PyObject* backward_function) { argument
1636 [backward_function]() {
1637 Py_INCREF(backward_function);
1639 new PyBackwardFunction([backward_function](PyObject* out_grads) {
1640 return PyObject_CallObject(backward_function, out_grads);
1644 [backward_function](PyBackwardFunction* py_backward_function) {
1645 Py_DECREF(backward_function);
2154 op_outputs](PyBackwardFunction* backward_function) { argument
[all …]
|