Searched refs:add_retains_grad_hook (Results 1 – 2 of 2) sorted by relevance
210 new_fn->add_retains_grad_hook(std::move(out), self.output_nr()); in update_tensor_hooks_on_new_gradfn()566 fn->add_retains_grad_hook( in retain_grad()
501 void add_retains_grad_hook( in add_retains_grad_hook() function