Searched refs:_test_sum_reduction_vs_numpy (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_reductions.py | 1742 …def _test_sum_reduction_vs_numpy(self, torch_fn, np_fn, device, dtype, with_keepdim=False, with_ex… member in TestReductions
1768 self._test_sum_reduction_vs_numpy(torch.sum, np.sum, device, dtype)
1769 self._test_sum_reduction_vs_numpy(torch.sum, np.sum, device, dtype, with_extremal=True)
1770 self._test_sum_reduction_vs_numpy(torch.sum, np.sum, device, dtype, with_keepdim=True)
1775 self._test_sum_reduction_vs_numpy(torch.nansum, np.nansum, device, dtype)
1776 … self._test_sum_reduction_vs_numpy(torch.nansum, np.nansum, device, dtype, with_extremal=True)
1777 self._test_sum_reduction_vs_numpy(torch.nansum, np.nansum, device, dtype, with_keepdim=True)
|