Searched refs:gradient_function (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradients.cc | 66 std::unique_ptr<GradientFunction>* gradient_function) const { in Lookup()
72 gradient_function->reset(iter->second(op)); in Lookup()
111 const string& op_type, GradientFunction* gradient_function,
175 const string& op_type, GradientFunction* gradient_function, in CallBackwardFunction() argument
179 if (gradient_function == nullptr) { in CallBackwardFunction()
186 return gradient_function->Compute(ctx_, output_gradients, result); in CallBackwardFunction()
227 GradientFunction* gradient_function, in RecordOperation() argument
241 [gradient_function]() -> GradientFunction* { return gradient_function; }, in RecordOperation()
|
| D | gradients.h | 83 std::unique_ptr<GradientFunction>* gradient_function) const;
151 GradientFunction* gradient_function,
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | custom_gradient_test.cc | 75 std::unique_ptr<GradientFunction> gradient_function( in ExpWithPassThroughGrad() local
77 tape.RecordOperation(inputs, exp_outputs, gradient_function.release()); in ExpWithPassThroughGrad()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 822 PyObject* gradient_function = nullptr; variable
937 if (gradient_function != nullptr) { in TFE_Py_RegisterGradientFunction()
938 Py_DECREF(gradient_function); in TFE_Py_RegisterGradientFunction()
941 gradient_function = nullptr; in TFE_Py_RegisterGradientFunction()
948 gradient_function = e; in TFE_Py_RegisterGradientFunction()
3267 PyObject_CallObject(gradient_function, callback_args.get()));
|
| D | function.py | 1056 gradient_function = (
1059 {"PartitionedCall": gradient_function,
1060 "StatefulPartitionedCall": gradient_function}):
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 2711 gradient_function = op._gradient_function # pylint: disable=protected-access
2712 if gradient_function:
2713 return gradient_function
|