Home
last modified time | relevance | path

Searched refs:_test_unique_with_expects (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_sort_and_select.py916 def _test_unique_with_expects( member in TestSortAndSelect
990 self._test_unique_with_expects(
1059 self._test_unique_with_expects(
H A Dtest_mps.py4115 …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…