Home
last modified time | relevance | path

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 Dtest_multiarray.py2381 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 …]