Home
last modified time | relevance | path

Searched refs:ComputeGradient (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/
H A Dtape.py41 out_grads = self._c_tape.ComputeGradient(ctx, flat_targets, flat_sources,
H A Dtape.cc57 Status s = self->ComputeGradient(ctx, target_tensors, in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dgradients.h166 Status ComputeGradient(
H A Dgradients.cc272 Status Tape::ComputeGradient( in ComputeGradient() function in tensorflow::gradients::Tape
292 TF_RETURN_IF_ERROR(GradientTape::ComputeGradient( in ComputeGradient()
H A Dtape.h173 Status ComputeGradient(
682 Status GradientTape<Gradient, BackwardFunction, TapeTensor>::ComputeGradient( in ComputeGradient() function
1016 return tape->ComputeGradient(vspace_, targets, sources, in ForwardpropFromTape()
H A Dgradients_test.cc125 return tape.ComputeGradient(ctx, in RecordOperationWithNullGradientFunctionModel()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/
H A Dcustom_gradient_test.cc78 TF_RETURN_IF_ERROR(tape.ComputeGradient(ctx, in ExpWithPassThroughGrad()
H A Dgrad_test_helper.cc124 TF_RETURN_IF_ERROR(tape.ComputeGradient(ctx, /*targets=*/temp_outputs, in BuildGradModel()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tfe_src.cc2860 status->status = tape_obj->tape->ComputeGradient(