Home
last modified time | relevance | path

Searched refs:ModuleInterface (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/test/jit/
Dtest_module_interface.py47 class ModuleInterface(nn.Module): class
52 proxy_mod: ModuleInterface
279 class ModuleInterface(nn.Module): class
287 proxy_mod: ModuleInterface
312 class ModuleInterface(nn.Module): class
324 proxy_mod: ModuleInterface
340 class ModuleInterface(nn.Module): class
348 proxy_mod: ModuleInterface
405 class ModuleInterface(nn.Module): class
431 proxy_mod: ModuleInterface
[all …]
Dtest_module_containers.py463 class ModuleInterface(torch.nn.Module): class
485 value: ModuleInterface = self.d[key]
497 submodule: ModuleInterface = self[key]
511 submodule: ModuleInterface = self.d[key]
526 class ModuleInterface(torch.nn.Module): class
548 value: ModuleInterface = self.l[idx]
560 submodule: ModuleInterface = self[idx]
574 submodule: ModuleInterface = self.l[idx]
Dtest_freezing.py1530 class ModuleInterface(torch.nn.Module): class
1547 value: ModuleInterface = self.d[key]
1564 value: ModuleInterface = self.l[idx]
1577 class ModuleInterface(torch.nn.Module): class
1591 impl: ModuleInterface
1611 class ModuleInterface(torch.nn.Module): class
1624 impl: ModuleInterface
/external/pytorch/test/package/package_a/
Dfake_interface.py6 class ModuleInterface(torch.nn.Module): class
35 proxy_mod: ModuleInterface