Searched refs:_test_reduction_function_with_numpy (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_reductions.py | 1696 def _test_reduction_function_with_numpy(self, torch_func, np_func, device, dtype, member in TestReductions
1725 … self._test_reduction_function_with_numpy(torch.count_nonzero, np.count_nonzero, device, dtype)
1726 …self._test_reduction_function_with_numpy(torch.count_nonzero, np.count_nonzero, device, dtype, Tru…
1761 self._test_reduction_function_with_numpy(torch_fn, np_fn, device, dtype,
|