Lines Matching refs:ref_input
4507 ref_input = input.detach().clone().contiguous().requires_grad_(True)
4514 ref_out = ref_pus(ref_ps(ref_input))
4520 self.assertEqual(input.grad, ref_input.grad)
4796 ref_input = input.detach().clone().contiguous().requires_grad_(True)
4807 ref_out = ref_bn(ref_input)
4815 self.assertEqual(input.grad, ref_input.grad)
4849 ref_input = input.detach().clone().contiguous()
4854 ref_out = ref_bn(ref_input)
4880 ref_input = input.detach().clone().contiguous().requires_grad_(True)
4886 ref_out = ref_mod(ref_input)
4893 self.assertEqual(input.grad, ref_input.grad)
8433 …ref_input = input.detach().clone().contiguous(memory_format=torch.contiguous_format).requires_grad…
8442 ref_out = ref_gn(ref_input)
8454 self.assertEqual(input.grad, ref_input.grad, atol=atol, rtol=rtol)
10196 ref_input = input.clone().cpu().detach().requires_grad_(True)
10198 ref_output = fn(ref_input, dtype=torch.float, dim=dim)
10204 ref_grad_input, = torch.autograd.grad(ref_output, ref_input,
10211 self.assertEqual(input.grad, ref_input.grad)