Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_async.py469 class TestModuleWrapper(nn.Module): class
481 self.checkTrace(TestModuleWrapper(), (torch.randn(5, 5),))