Searched defs:test_nn_module (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/jit/ | ||
| H A D | test_pdt.py | 37 def test_nn_module(self): member in TestPDT |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_utils.py | 1125 def test_nn_module(self): member in TestDeviceUtils |
| H A D | test_legacy_vmap.py | 821 def test_nn_module(self): member in TestVmapAPILegacy |
| /aosp_15_r20/external/pytorch/test/dynamo/ | ||
| H A D | test_modules.py | 1681 def test_nn_module(self): member in OptimizedModuleTest |
| /aosp_15_r20/external/pytorch/test/functorch/ | ||
| H A D | test_vmap.py | 1054 def test_nn_module(self): member in TestVmapAPI |