Searched refs:backward_function (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | tape.py | 177 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
181 input_tensors, backward_function,
186 backward_function): argument
190 backward_function)
194 backward_function, argument
211 op_type, output_tensors, input_tensors, backward_function,
|
| D | function.py | 541 backward_function = ConcreteFunction(
545 backward_function.name})
550 return forward_function, backward_function
641 forward_function, backward_function = _create_forward_backward_with_graph(
643 return forward_function, backward_function
749 backward_function, to_record = self._backward(flat_outputs)
752 backward_function)
865 forward_function, backward_function = _create_forward_backward_with_graph(
871 return (forward_function, self._func_graph, backward_function,
875 forward_function, backward_function, inference_args, input_tangents)
[all …]
|
| D | pywrap_tfe.h | 217 PyObject* backward_function,
226 PyObject* backward_function);
244 PyObject* backward_function, PyObject* forwardprop_output_indices);
|
| D | pywrap_tfe_src.cc | 1370 const string& op_type, PyBackwardFunction* backward_function, in CallBackwardFunction() argument
1384 PyObject* py_result = (*backward_function)(grads, unneeded_gradients); in CallBackwardFunction()
2621 PyObject* backward_function, argument
2634 [backward_function]() {
2635 Py_INCREF(backward_function);
2637 [backward_function](PyObject* out_grads,
2639 return PyObject_CallObject(backward_function, out_grads);
2644 [backward_function](PyBackwardFunction* py_backward_function) {
2645 Py_DECREF(backward_function);
2676 PyObject* backward_function, PyObject* forwardprop_output_indices) { argument
[all …]
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 46 BackwardFunction* backward_function; member
100 const string& op_type, BackwardFunction* backward_function,
135 pair.second.backward_function_deleter(pair.second.backward_function); in ~GradientTape()
469 op_it->second.backward_function_deleter(op_it->second.backward_function); in DeleteTrace()
576 op_pair.second.backward_function); in PrepareBackprop()
703 pair.second.backward_function_deleter(pair.second.backward_function); in ComputeGradient()
784 s = vspace.CallBackwardFunction(trace.op_type, trace.backward_function, in ComputeGradient()
788 trace.backward_function_deleter(trace.backward_function); in ComputeGradient()
795 trace.backward_function_deleter(trace.backward_function); in ComputeGradient()
981 backward_function(backward_function_getter(), in ForwardpropFromTape()
[all …]
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 1065 const py::handle& input_tensors, const py::handle& backward_function, in PYBIND11_MODULE()
1069 backward_function.ptr(), forward_function.ptr())); in PYBIND11_MODULE()
1074 const py::handle& input_tensors, const py::handle& backward_function) { in PYBIND11_MODULE() argument
1077 backward_function.ptr())); in PYBIND11_MODULE()
1082 const py::handle& input_tensors, const py::handle& backward_function, in PYBIND11_MODULE()
1086 backward_function.ptr(), forwardprop_output_indices.ptr())); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | func_graph.py | 822 backward_function=lambda x: [x],
996 backward_function=lambda x: [x],
1015 backward_function=lambda x: [x],
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 713 backward_function=lambda x: [x],
|