Searched refs:backward_function (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | tape.py | 181 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
185 input_tensors, backward_function,
190 backward_function): argument
194 backward_function)
198 backward_function, argument
215 op_type, output_tensors, input_tensors, backward_function,
|
| D | pywrap_tfe.h | 213 PyObject* backward_function,
222 PyObject* backward_function);
240 PyObject* backward_function, PyObject* forwardprop_output_indices);
|
| D | function.py | 636 backward_function = ConcreteFunction(
640 backward_function.name})
645 return forward_function, backward_function
736 forward_function, backward_function = _create_forward_backward_with_graph(
738 return forward_function, backward_function
845 backward_function, to_record = self._backward(flat_outputs)
848 backward_function)
961 forward_function, backward_function = _create_forward_backward_with_graph(
967 return (forward_function, self._func_graph, backward_function,
971 forward_function, backward_function, inference_args, input_tangents)
[all …]
|
| D | pywrap_tfe_src.cc | 1331 const string& op_type, PyBackwardFunction* backward_function, in CallBackwardFunction() argument
1345 PyObject* py_result = (*backward_function)(grads, unneeded_gradients); in CallBackwardFunction()
2549 PyObject* backward_function, argument
2562 [backward_function]() {
2563 Py_INCREF(backward_function);
2565 [backward_function](PyObject* out_grads,
2567 return PyObject_CallObject(backward_function, out_grads);
2572 [backward_function](PyBackwardFunction* py_backward_function) {
2573 Py_DECREF(backward_function);
2604 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 | 906 const py::handle& input_tensors, const py::handle& backward_function, in PYBIND11_MODULE()
910 backward_function.ptr(), forward_function.ptr())); in PYBIND11_MODULE()
915 const py::handle& input_tensors, const py::handle& backward_function) { in PYBIND11_MODULE() argument
918 backward_function.ptr())); in PYBIND11_MODULE()
923 const py::handle& input_tensors, const py::handle& backward_function, in PYBIND11_MODULE()
927 backward_function.ptr(), forwardprop_output_indices.ptr())); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | func_graph.py | 667 backward_function=lambda x: [x],
718 backward_function=lambda x: [x],
739 backward_function=lambda x: [x],
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 682 backward_function=lambda x: [x],
|