Searched refs:_test_unique_with_expects (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_sort_and_select.py | 916 def _test_unique_with_expects( member in TestSortAndSelect 990 self._test_unique_with_expects( 1059 self._test_unique_with_expects(
|
H A D | test_mps.py | 4115 …def _test_unique_with_expects(self, device, dtype, f, x, expected_unique, expected_inverse, expect… member in TestMPS 4166 …self._test_unique_with_expects(device, dtype, f, x, expected_unique, expected_inverse, expected_co…
|