Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_freezing.py138 class SubModule2(nn.Module): class
152 self.sub2 = SubModule2()
243 class SubModule2(nn.Module): class
258 self.sub = SubModule2()
393 class SubModule2(nn.Module): class
407 self.sub2 = SubModule2()
487 class SubModule2(nn.Module): class
499 self.sub2 = SubModule2()
546 class SubModule2(nn.Module): class
558 self.sub2 = SubModule2()
/aosp_15_r20/external/pytorch/test/
H A Dtest_static_runtime.py168 class SubModule2(nn.Module): class
183 self.sub2 = SubModule2()