Lines Matching refs:ref_input

227             ref_input = input.detach().clone().contiguous().requires_grad_(True)
233             ref_out = ref_pool(ref_input)
239             self.assertEqual(input.grad, ref_input.grad)
254             ref_input = input.detach().clone().contiguous().requires_grad_(True)
260             ref_out = ref_pool(ref_input)
266             self.assertEqual(input.grad, ref_input.grad)
324         ref_input = input.detach().clone().contiguous().requires_grad_(True)
330         ref_out = ref_pool(ref_input)
336         self.assertEqual(input.grad, ref_input.grad)
347         ref_input = input.detach().clone().contiguous().requires_grad_(True)
351         ref_out = ref_pool(ref_input)
407         ref_input = input.clone().contiguous()
413         ref_out, ref_ind = ref_pool(ref_input)
963             ref_input = input.detach().clone().contiguous().requires_grad_(True)
974             ref_out = ref_pool(ref_input)
980             self.assertEqual(input.grad, ref_input.grad)
1097             ref_input = input.detach().clone().contiguous().requires_grad_(True)
1105             ref_out, ref_ind = ref_pool(ref_input)
1114             self.assertEqual(input.grad, ref_input.grad)
1187             ref_input = input.detach().clone().contiguous().requires_grad_(True)
1194             ref_out, ref_ind = ref_pool(ref_input)
1214                 self.assertEqual(input.grad, ref_input.grad, atol=0.05, rtol=0.01)
1216                 self.assertEqual(input.grad, ref_input.grad)
1353             ref_input = input.detach().clone().contiguous().requires_grad_(True)
1359             ref_out, ref_ind = ref_pool(ref_input)
1370             self.assertEqual(input.grad, ref_input.grad)
1414             ref_input = input.detach().clone().contiguous().requires_grad_(True)
1427             ref_out, ref_ind = ref_pool(ref_input)
1436             self.assertEqual(input.grad, ref_input.grad)