Searched refs:DeleteGradient (Results 1 – 3 of 3) sorted by relevance
120 virtual void DeleteGradient(Gradient* gradient) const = 0;241 vspace_.DeleteGradient(accumulated.second); in ~ForwardAccumulator()253 void DeleteGradient(int64 tensor_id);799 vspace.DeleteGradient(grad); in ComputeGradient()841 vspace.DeleteGradient(g); in ComputeGradient()892 vspace.DeleteGradient(g); in ComputeGradient()975 this->vspace_.DeleteGradient(tensor); in ForwardpropFromTape()1055 this->vspace_.DeleteGradient(tensor); in Accumulate()1106 vspace_.DeleteGradient(forward_grads[i]); in Accumulate()1134 void ForwardAccumulator<Gradient, BackwardFunction, TapeTensor>::DeleteGradient( in DeleteGradient() function[all …]
128 void DeleteGradient(AbstractTensorHandle* gradient) const override;218 void TapeVSpace::DeleteGradient(AbstractTensorHandle* gradient) const { in DeleteGradient() function in tensorflow::gradients::TapeVSpace
1379 void DeleteGradient(PyObject* tensor) const final { Py_XDECREF(tensor); } in DeleteGradient() function in PyVSpace2269 accumulator->accumulator->DeleteGradient(parsed_tensor_id);