Home
last modified time | relevance | path

Searched refs:tensor_post_acc_grad_hooks (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/autograd/functions/
Daccumulate_grad.cpp60 auto& hook = tensor_post_acc_grad_hooks(); in apply()
73 auto& hook = tensor_post_acc_grad_hooks(); in compiled_args()
96 auto& hook = tensor_post_acc_grad_hooks(); in apply_with_saved()
Daccumulate_grad.h53 std::unique_ptr<PostAccumulateGradHook>& tensor_post_acc_grad_hooks() noexcept in tensor_post_acc_grad_hooks() function
/external/pytorch/torch/csrc/autograd/
Dfunction.h528 tensor_post_acc_grad_hooks() noexcept { in tensor_post_acc_grad_hooks() function