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 …]
|