Searched refs:tensor_post_acc_grad_hooks (Results 1 – 3 of 3) sorted by relevance
60 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()
53 std::unique_ptr<PostAccumulateGradHook>& tensor_post_acc_grad_hooks() noexcept in tensor_post_acc_grad_hooks() function
528 tensor_post_acc_grad_hooks() noexcept { in tensor_post_acc_grad_hooks() function