Searched defs:test_hooks (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/nn/ | ||
| H A D | test_module_hooks.py | 1307 def test_hooks(self): member in TestModuleHookNN |
| /aosp_15_r20/external/pytorch/test/dynamo/ | ||
| H A D | test_higher_order_ops.py | 2223 def test_hooks(self): member in HigherOrderOpTests |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_autograd.py | 1734 def test_hooks(self): member in TestAutograd |