Home
last modified time | relevance | path

Searched refs:ModWithTensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/package/
H A Dtest_package_script.py180 from package_a.test_module import ModWithTensor
182 scripted_mod = torch.jit.script(ModWithTensor(torch.rand(1, 2, 3)))
202 from package_a.test_module import ModWithTensor
204 scripted_mod = torch.jit.script(ModWithTensor(torch.rand(1, 2, 3)))
219 from package_a.test_module import ModWithSubmod, ModWithTensor
222 ModWithSubmod(ModWithTensor(torch.rand(1, 2, 3)))
285 from package_a.test_module import ModWithTensor, SimpleTest
288 scripted_mod_1 = torch.jit.script(ModWithTensor(torch.rand(1, 2, 3)))
312 ModWithTensor,
317 scripted_mod_1 = torch.jit.script(ModWithTensor(torch.rand(1, 2, 3)))
[all …]
H A Dtest_directory_reader.py268 from package_a.test_module import ModWithTensor
270 scripted_mod = torch.jit.script(ModWithTensor(torch.rand(1, 2, 3)))
/aosp_15_r20/external/pytorch/test/package/package_a/
H A Dtest_module.py19 class ModWithTensor(torch.nn.Module): class