Home
last modified time | relevance | path

Searched defs:output_gradients (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dimperative_grad.py32 output_gradients=None, argument
H A Dbackprop.py1010 output_gradients=None, argument
H A Dpywrap_tfe_src.cc1372 tensorflow::gtl::ArraySlice<PyObject*> output_gradients, in CallBackwardFunction()
2797 PyObject* sources, PyObject* output_gradients,
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/
H A Dtape.py37 def gradient(self, targets, sources, output_gradients=None): argument
H A Dtape.cc55 std::vector<AbstractTensorHandle*> output_gradients) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dtape.h601 gtl::ArraySlice<Gradient*> output_gradients, const TensorTape& tensor_tape, in InitialGradients()
687 gtl::ArraySlice<Gradient*> output_gradients, absl::Span<Gradient*> result, in ComputeGradient()
H A Dgradients.cc177 gtl::ArraySlice<AbstractTensorHandle*> output_gradients, in CallBackwardFunction()
275 absl::Span<AbstractTensorHandle* const> output_gradients, in ComputeGradient()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dpooling_ops_3d_test.py30 def _AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding): argument
H A Dpooling_ops_test.py550 def AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dtfe_wrapper.cc1092 const py::handle& unconnected_gradients) { in PYBIND11_MODULE()