Lines Matching full:ones

92         x = torch.ones(1, requires_grad=True)
93         y = torch.ones(1, requires_grad=True)
110         x = torch.ones([2, 2], requires_grad=True)
111         y = torch.ones([1], requires_grad=True)
114         b.backward(torch.ones([2, 2], dtype=torch.float), retain_graph=True)
115         b.backward(torch.ones([2, 2], dtype=torch.float))
116         # warmup_backward(b, torch.ones([2, 2], dtype=torch.float))
151         x = torch.ones(1)
152         y = torch.ones(1)
193         x = torch.ones(1)
198         x = torch.ones([2, 3, 5])
209         x = torch.ones(1)
210         y = torch.ones(1)
226         x = torch.ones(1, requires_grad=True)
227         y = torch.ones(1, requires_grad=True)
230         b.backward(torch.ones([1], dtype=torch.float), retain_graph=True)
231         b.backward(torch.ones([1], dtype=torch.float))
240         x = torch.ones(1, requires_grad=False)
276         x = torch.ones(1, requires_grad=False)