Searched refs:_test_reduce_integer_upcast (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_reductions.py | 888 def _test_reduce_integer_upcast(self, fn, has_out=True, test_complex=True): member in TestReductions
936 self._test_reduce_integer_upcast(lambda x, **kwargs: torch.sum(x, **kwargs), False)
937 self._test_reduce_integer_upcast(lambda x, **kwargs: torch.sum(x, 0, **kwargs))
941 self._test_reduce_integer_upcast(lambda x, **kwargs: torch.prod(x, **kwargs), False)
942 self._test_reduce_integer_upcast(lambda x, **kwargs: torch.prod(x, 0, **kwargs))
946 self._test_reduce_integer_upcast(lambda x, **kwargs: torch.cumsum(x, 0, **kwargs))
950 self._test_reduce_integer_upcast(lambda x, **kwargs: torch.cumprod(x, 0, **kwargs))
|