Searched refs:MarkAsResult (Results 1 – 3 of 3) sorted by relevance
117 virtual void MarkAsResult(Gradient* gradient) const = 0;771 vspace.MarkAsResult(new_gradients); in ComputeGradient()1010 vspace_.MarkAsResult(in_grad); in ForwardpropFromTape()1120 vspace_.MarkAsResult(tangent); in Watch()
126 void MarkAsResult(AbstractTensorHandle* gradient) const override;216 void TapeVSpace::MarkAsResult(AbstractTensorHandle* gradient) const {} in MarkAsResult() function in tensorflow::gradients::TapeVSpace
1280 void MarkAsResult(PyObject* gradient) const final { Py_INCREF(gradient); } in MarkAsResult() function in PyVSpace