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],
|