Lines Matching full:ones_like

142                 self.assertEqual(x.grad, torch.ones_like(x))
159                     self.assertEqual(x.grad, torch.ones_like(x))
169                     self.assertEqual(x.grad, torch.ones_like(x))
243                         y, x, torch.ones_like(y), retain_graph=True
247                 expected = torch.ones_like(x)
250                     (gx,) = torch.autograd.grad(z, x, torch.ones_like(z))
301                         torch.autograd.grad(z, y, torch.ones_like(z), allow_unused=True)
310                         torch.autograd.grad(z, x, torch.ones_like(z), allow_unused=True)
409                     x, (a, b), torch.ones_like(x), allow_unused=True, retain_graph=True
414                     y, (a, b), torch.ones_like(y), allow_unused=True, retain_graph=True
419                     z, (a, b), torch.ones_like(z), allow_unused=True, retain_graph=True
443                     torch.ones_like(a),
451                     torch.ones_like(b),