Lines Matching full:ones_like

169         self.assertEqual(old_grad_weight, torch.ones_like(m.weight))
171         self.assertEqual(old_grad_bias, torch.ones_like(m.bias))
202         self.assertEqual(old_grad_weight, torch.ones_like(m.weight))
204         self.assertEqual(old_grad_bias, torch.ones_like(m.bias))
213             compute_mask.return_value = torch.ones_like(m.weight)
237         mask = torch.ones_like(m.weight)
252         self.assertEqual(m.bias.grad, torch.ones_like(m.bias))
292         mask = torch.ones_like(m.weight)
510         computed_mask = p.compute_mask(t, default_mask=torch.ones_like(t))
521         computed_mask = p.compute_mask(t, default_mask=torch.ones_like(t))
540         expected_mask_axis1 = torch.ones_like(m.weight)
575         expected_mask_axis1 = torch.ones_like(m.weight)