Searched refs:assert_partitioned (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/ |
H A D | test_multiarray.py | 2381 def assert_partitioned(self, d, kth): member in TestMethods 2405 self.assert_partitioned(p, (0, 3)) 2406 self.assert_partitioned(d[np.argpartition(d, (0, 3))], (0, 3)) 2424 self.assert_partitioned(p, keys) 2426 self.assert_partitioned(p, keys) 2433 self.assert_partitioned(np.partition(d, [5] * 4), [5]) 2434 self.assert_partitioned(np.partition(d, [5] * 4 + [6, 13]), [5] * 4 + [6, 13]) 2435 self.assert_partitioned(d[np.argpartition(d, [5] * 4)], [5]) 2436 self.assert_partitioned( 2451 self.assert_partitioned(p[i, :], kth) [all …]
|