Searched refs:checkGraphModule (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fx.py | 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)])
[all …]
|