Lines Matching full:ones

82         f.set_result(torch.ones(2, 2))
93         f1.set_result(torch.ones(2, 2))
103         f.set_result(torch.ones(2, 2))
105         self.assertEqual(f.wait(), torch.ones(2, 2))
115         t = threading.Thread(target=slow_set_future, args=(f, torch.ones(2, 2)))
118         self.assertEqual(f.wait(), torch.ones(2, 2))
141         fut.set_result(torch.ones(2, 2))
142         self.assertEqual(fut.wait(), torch.ones(2, 2))
143         self.assertEqual(then_fut.wait(), torch.ones(2, 2) + 1)
153         fut.set_result(torch.ones(2, 2))
156             self.assertEqual(futs[i].wait(), torch.ones(2, 2) + i + 1)
200         fut.set_result(torch.ones(2, 2))
201         self.assertEqual(fut.wait(), torch.ones(2, 2))
221         fut.set_result(torch.ones(2, 2))
222         self.assertEqual(fut.wait(), torch.ones(2, 2))
272         fut.set_result(torch.ones(2, 2))
273         self.assertEqual(fut.wait(), torch.ones(2, 2))
275         self.assertEqual(then_fut.wait(), torch.ones(2, 2) + 1)
286         fut.set_result(torch.ones(2, 2))
290         self.assertEqual(fut.wait(), torch.ones(2, 2))