Lines Matching full:ones

87         c2p.put((rank, torch.ones(2, 2) * 2, xs[0].to("cpu")))
99             c2p.put((rank, torch.ones(2, 2) * i, ys[0][i].to("cpu")))
132         x = torch.ones(5, 5, device=device) + self.rank
135         self.assertEqual(y, 1 + torch.ones(5, 5))
152         x = torch.ones(5, 5, device=device) + self.rank
157             self.assertEqual(y, 3 * torch.ones(5, 5, device=device))
162         x_g = (3 * torch.ones(5, 5, device=device)).cos()
173         x = torch.ones(5, 5, device=device) + self.rank
177         self.assertEqual(y, 3 * torch.ones(5, 5, device=device))
181         x_g = 2 * (3 * torch.ones(5, 5, device=device)).cos()
192         x = torch.ones(5, 5, device=device) + self.rank
196             self.assertEqual(t, torch.ones(5, 5, device=device) + i)
201         x_s = 2 * (3 * torch.ones(5, 5, device=device)).cos()
212         x0 = torch.ones(5, 5, device=device) + 2 * self.rank
213         x1 = torch.ones(5, 5, device=device) + 2 * self.rank
220             self.assertEqual(t, torch.ones(5, 5, device=device) + 2 * i)
224         x_s = (4 * torch.ones(5, 5, device=device)).cos()
237         x = torch.ones(int(row), 5, device=device) * (self.rank + 1)
251         x_s = ((self.rank + 1) * torch.ones(int(row), 5, device=device)).cos()