Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_module_hooks.py1221 def _test_hooks(self, backward_register_fn): member in TestModuleHookNN
1308 self._test_hooks("register_backward_hook")
1309 self._test_hooks("register_full_backward_hook")
1310 self._test_hooks("register_full_backward_pre_hook")