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