Searched refs:compare_with_numpy (Results 1 – 9 of 9) sorted by relevance
172 self.compare_with_numpy(212 self.compare_with_numpy(220 self.compare_with_numpy(torch_fn, np_fn, x, device=None, dtype=None)224 self.compare_with_numpy(torch_fn, np_fn, x, device=None, dtype=None)229 self.compare_with_numpy(torch_fn, np_fn, x, device=None, dtype=None)569 self.compare_with_numpy(torch_fn, np_fn, data)581 self.compare_with_numpy(torch_fn, np_fn, t_in)592 self.compare_with_numpy(torch_fn, np_fn, data)658 self.compare_with_numpy(torch_fn, np_fn, data)
472 self.compare_with_numpy(477 self.compare_with_numpy(540 self.compare_with_numpy(torch.sqrt, np.sqrt, x)545 self.compare_with_numpy(torch.acos, np.arccos, x)552 self.compare_with_numpy(torch.sqrt, np.sqrt, x)584 self.compare_with_numpy(torch.digamma, scipy.special.digamma, x)608 self.compare_with_numpy(torch.digamma, scipy.special.digamma, tensor)1548 self.compare_with_numpy(torch.exp, np.exp, a)1568 self.compare_with_numpy(torch.exp, np.exp, inf_real_zero_imag_in)1577 self.compare_with_numpy(torch.exp, np.exp, zero_real_inf_imag_in)[all …]
1160 self.compare_with_numpy(torchfn, reffn, x)1164 self.compare_with_numpy(torchfn, reffn, x)1178 …self.compare_with_numpy(lambda x: get_values(torchfn(x, d, False)), lambda x: reffn(x, d, keepdims…1710 self.compare_with_numpy(torch_func, np_func, x, device=None, dtype=None,1720 … self.compare_with_numpy(torch_func_partial, np_func_partial, x, device=None, dtype=None,1798 … self.compare_with_numpy(torch_fn, np_fn, x, device=None, dtype=None, atol=atol, rtol=rtol)1804 self.compare_with_numpy(torch.argmax, np.argmax, t)1805 self.compare_with_numpy(torch.argmin, np.argmin, t)1810 self.compare_with_numpy(torch.argmax, np.argmax, t)1811 self.compare_with_numpy(torch.argmin, np.argmin, t)[all …]
448 def compare_with_numpy(contiguous_input=True): function468 compare_with_numpy()469 compare_with_numpy(contiguous_input=False)1524 self.compare_with_numpy(1537 self.compare_with_numpy(torch_fn, np_fn, x, device=None, dtype=None)1549 self.compare_with_numpy(
3114 self.compare_with_numpy(3170 self.compare_with_numpy(lambda x: x << 2, lambda x: np.left_shift(x, 2), a)3175 self.compare_with_numpy(lambda x: x >> 1, lambda x: np.right_shift(x, 1), a)3205 self.compare_with_numpy(3214 self.compare_with_numpy(4234 self.compare_with_numpy(4237 self.compare_with_numpy(4240 self.compare_with_numpy(4287 self.compare_with_numpy(
909 self.compare_with_numpy(op, np.linalg.det, t)6768 self.compare_with_numpy(torch_fn, np_fn, sx)6771 self.compare_with_numpy(torch_fn, np_fn, sx[0])
1619 self.compare_with_numpy(torch.exp, np.exp, a)
720 self.compare_with_numpy(torch_fn, np_fn, sx)723 self.compare_with_numpy(torch_fn, np_fn, sx[0])
3751 def compare_with_numpy(self, torch_fn, np_fn, tensor_like, member in TestCase