Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_backends.py118 def test_save_load(self): member in JitBackendTestCase
154 def test_save_load(self): member in BasicModuleTest
213 def test_save_load(self): member in BasicModuleUnavailableTest
273 def test_save_load(self): member in NestedModuleTest
359 def test_save_load(self): member in SelectiveLoweringTest
488 def test_save_load(self): member in TestBackends
489 self.basic_module_test.test_save_load()
490 self.basic_module_unavailable_test.test_save_load()
491 self.nested_module_test.test_save_load()
492 self.selective_lowering_test.test_save_load()
H A Dtest_generator.py128 def test_save_load(self): member in TestGenerator
H A Dtest_cuda.py643 def test_save_load(self): function
/aosp_15_r20/external/pytorch/test/
H A Dtest_package.py22 from package.test_save_load import TestSaveLoad # noqa: F401
H A Dtest_modules.py215 def test_save_load(self, device, dtype, module_info, training): member in TestModule
H A Dtest_jit.py33 from jit.test_save_load import TestSaveLoad, TestSaveLoadFlatbuffer # noqa: F401
/aosp_15_r20/external/pytorch/test/custom_backend/
H A Dtest_custom_backend.py33 def test_save_load(self): member in TestCustomBackend
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_config.py43 def test_save_load(self): member in TestInductorConfig
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A DCMakeLists.txt82 ${JIT_TEST_ROOT}/test_save_load.cpp
/aosp_15_r20/external/pytorch/test/quantization/pt2e/
H A Dtest_quantize_pt2e.py1459 def test_save_load(self): member in TestQuantizePT2E