Searched refs:ModWithTensor (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/test/package/ |
H A D | test_package_script.py | 180 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 D | test_directory_reader.py | 268 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 D | test_module.py | 19 class ModWithTensor(torch.nn.Module): class
|