Searched refs:_test_hooks (Results 1 – 1 of 1) sorted by relevance
1221 def _test_hooks(self, backward_register_fn): member in TestModuleHookNN1308 self._test_hooks("register_backward_hook")1309 self._test_hooks("register_full_backward_hook")1310 self._test_hooks("register_full_backward_pre_hook")