Searched refs:assertExportImportModule (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_jit.py | 150 self.assertExportImportModule(m, inputs)
152 def assertExportImportModule(self, m, inputs): member in JitCommonTestCase
|
| H A D | jit_metaprogramming_utils.py | 620 self.assertExportImportModule(module, tensors)
|
| H A D | jit_utils.py | 641 self.assertExportImportModule(sm, args)
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_module_containers.py | 194 self.assertExportImportModule(m, (torch.randn(2, 2),))
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 9339 self.assertExportImportModule(m, (torch.randn(2, 2),))
13123 self.assertExportImportModule(strong_mod, (inp,))
16118 self.assertExportImportModule(module, tensors)
|