Searched refs:PyFunctionTensorPostAccGradHooks (Results 1 – 3 of 3) sorted by relevance
205 PyFunctionTensorPostAccGradHooks::PyFunctionTensorPostAccGradHooks( in PyFunctionTensorPostAccGradHooks() function in torch::autograd::PyFunctionTensorPostAccGradHooks212 PyFunctionTensorPostAccGradHooks::~PyFunctionTensorPostAccGradHooks() { in ~PyFunctionTensorPostAccGradHooks()220 auto PyFunctionTensorPostAccGradHooks::operator()(const Variable& tensor) in operator ()()230 void PyFunctionTensorPostAccGradHooks::compiled_args( in compiled_args()241 void PyFunctionTensorPostAccGradHooks::apply_with_saved( in apply_with_saved()
44 struct PyFunctionTensorPostAccGradHooks : public PostAccumulateGradHook { struct45 PyFunctionTensorPostAccGradHooks(PyObject* dict);46 ~PyFunctionTensorPostAccGradHooks() override;
1345 tensor, std::make_unique<PyFunctionTensorPostAccGradHooks>(obj)); in THPVariable_set_post_accumulate_grad_hooks()