Lines Matching refs:MyModule
278 class MyModule(torch.nn.Module): class
327 m = MyModule()
345 class MyModule(torch.nn.Module): class
356 m = MyModule()
361 class MyModule(torch.nn.Module): class
375 m = MyModule()
440 class MyModule(torch.nn.Module): class
453 m = MyModule()
712 class MyModule(nn.Module): class
725 self.checkModule(MyModule(), (torch.zeros(1)))
728 class MyModule(nn.Module): class
742 self.checkModule(MyModule(), (torch.zeros(1),))
745 class MyModule(nn.Module): class
758 self.checkModule(MyModule(), (torch.ones(1),))