Home
last modified time | relevance | path

Searched refs:ModuleForwardSingleInput (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_hooks.py28 ModuleForwardSingleInput,
119 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
137 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
156 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
178 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
201 m_submod_call = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
230 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
250 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
262 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
274 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
[all …]
H A Dtest_hooks_modules.py41 class ModuleForwardSingleInput(torch.nn.Module): class
203 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
224 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
247 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
265 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
364 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
385 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
404 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
426 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
444 m = ModuleForwardSingleInput("outer_mod_name", "inner_mod_name")
[all …]