Searched refs:test_nn_module (Results 1 – 7 of 7) sorted by relevance
16 from package_a.test_nn_module import TestNnModule18 test_nn_module = TestNnModule()23 pe1.save_pickle("nn_module", "nn_module.pkl", test_nn_module)
6 from package.package_a.test_nn_module import TestNnModule # noqa: F401
1125 def test_nn_module(self): member in TestDeviceUtils
821 def test_nn_module(self): member in TestVmapAPILegacy
37 def test_nn_module(self): member in TestPDT
1681 def test_nn_module(self): member in OptimizedModuleTest
1054 def test_nn_module(self): member in TestVmapAPI