Lines Matching refs:checkGraphModule

183     def checkGraphModule(self, m: torch.nn.Module, args, kwargs=None):  member in TestFX
275         self.checkGraphModule(t, (torch.rand(1), torch.rand(1)), {'foo': torch.rand(1)})
300             self.checkGraphModule(t, (torch.rand(1), torch.rand(1)), {'foo': torch.rand(1)})
310         self.checkGraphModule(m, (input,))
320         self.checkGraphModule(m, (input,))
330         self.checkGraphModule(m, (input_dict,))
733         self.checkGraphModule(m, (a, b))
1305         self.checkGraphModule(Sqrt1(), [torch.zeros(8)])
1306         self.checkGraphModule(Sqrt2(), [torch.zeros(8)])
1307         self.checkGraphModule(Sqrt3(), [torch.zeros(8)])
2125         self.checkGraphModule(m, (torch.rand(3, 4),))
2160         self.checkGraphModule(fm, (torch.rand(3, 4),))
2433         self.checkGraphModule(foo, (x, y))
2598         self.checkGraphModule(m, ())
2599         self.checkGraphModule(m, (3,))
2607         self.checkGraphModule(m, ())
2608         self.checkGraphModule(m, (3,))
2609         self.checkGraphModule(m, (3, 4))
2617         self.checkGraphModule(m, (2,))
2618         self.checkGraphModule(m, (2, 3))
2626         self.checkGraphModule(m, ())
2656         self.checkGraphModule(m, ())
2772         self.checkGraphModule(GetItem1(), [torch.zeros(4)])
2773         self.checkGraphModule(GetItem2(), [torch.zeros(4)])
2774         self.checkGraphModule(GetItem3(), [torch.zeros(4)])
3493         self.checkGraphModule(M(), (torch.rand(2, 3), A()), kwargs=None)
3504         self.checkGraphModule(M(), (torch.rand(2, 3), A()), kwargs=None)
3515         self.checkGraphModule(M(), (torch.rand(2, 3), A()), kwargs=None)
3526         self.checkGraphModule(M(), (torch.rand(2, 3), A()), kwargs=None)