Lines Matching full:allclose

45         self.assertTrue(torch.allclose(torch.eye(2) + 102, script_out))
46         self.assertTrue(torch.allclose(script_out, out))
60         self.assertTrue(torch.allclose(torch.eye(2) + 13, script_out))
61         self.assertTrue(torch.allclose(script_out, out))
84         self.assertTrue(torch.allclose(torch.eye(2), script_out))
85         self.assertTrue(torch.allclose(script_out, out))
113         self.assertTrue(torch.allclose(torch.eye(2), script_out))
114         self.assertTrue(torch.allclose(script_out, out))
142         self.assertTrue(torch.allclose(torch.eye(2) + 2 * torch.ones(2), script_out))
143         self.assertTrue(torch.allclose(script_out, out))
174         self.assertTrue(torch.allclose(torch.eye(2) + 6 * torch.ones(2), script_out))
175         self.assertTrue(torch.allclose(script_out, out))
212         self.assertTrue(torch.allclose(torch.eye(2) + 7 * torch.ones(2), script_out))
213         self.assertTrue(torch.allclose(script_out, out))
244         self.assertTrue(torch.allclose(6 * torch.eye(2), script_out))
245         self.assertTrue(torch.allclose(script_out, out))
280             torch.allclose(2 * torch.eye(2) + 2 * torch.ones(2), script_out)
282         self.assertTrue(torch.allclose(script_out, out))
310             torch.allclose(2 * torch.eye(2) + 2 * torch.ones(2), script_out)
312         self.assertTrue(torch.allclose(script_out, out))
352         self.assertTrue(torch.allclose(expected, script_out))
353         self.assertTrue(torch.allclose(script_out, out))
360         self.assertTrue(torch.allclose(expected, script_out_load))
384         self.assertTrue(torch.allclose(expected, script_out))
385         self.assertTrue(torch.allclose(script_out, out))
392         self.assertTrue(torch.allclose(expected, script_out_load))