| /dports/math/xtensor/xtensor-0.24.0/test/ |
| H A D | test_xsort.cpp | 24 TEST(xsort, argsort) in TEST() argument
35 EXPECT_EQ(ex, argsort(a1, 0)); in TEST()
36 EXPECT_EQ(ex, argsort(a1)); in TEST()
41 EXPECT_EQ(ex2_1, argsort(a2_c)); in TEST()
42 EXPECT_EQ(ex2_1, argsort(a2_r)); in TEST()
52 EXPECT_EQ(ex3_2, argsort(a3)); in TEST()
53 EXPECT_EQ(ex3_0, argsort(a3, 0)); in TEST()
54 EXPECT_EQ(ex3_1, argsort(a3, 1)); in TEST()
55 EXPECT_EQ(ex3_2, argsort(a3, 2)); in TEST()
59 EXPECT_EQ(ex4, argsort(t1, 0)); in TEST()
[all …]
|
| /dports/math/py-numpy/numpy-1.20.3/numpy/ma/tests/ |
| H A D | test_deprecations.py | 11 def _test_base(self, argsort, cls): argument
13 argsort(arr_0d)
16 argsort(arr_1d)
21 np.ma.core.MaskedArrayFutureWarning, argsort, arr_2d)
22 assert_equal(result, argsort(arr_2d, axis=None))
25 argsort(arr_2d, axis=None)
26 argsort(arr_2d, axis=-1)
29 return self._test_base(np.ma.argsort, np.ndarray)
32 return self._test_base(np.ma.argsort, np.ma.MaskedArray)
35 return self._test_base(np.ma.MaskedArray.argsort, np.ma.MaskedArray)
|
| /dports/misc/openvdb/openvdb-9.0.0/openvdb_ax/openvdb_ax/test/snippets/function/ |
| H A D | argsort | 11 sorted3d = argsort(input3d);
14 sorted3f = argsort(input3f);
17 sorted3i = argsort(input3i);
20 sorted4d = argsort(input4d);
23 sorted4f = argsort(input4f);
26 sorted4i = argsort(input4i);
|
| /dports/math/py-pandas/pandas-1.2.5/pandas/tests/series/methods/ |
| H A D | test_argsort.py | 26 argsorted = datetime_series.argsort()
36 result = s.argsort()
40 result = shifted.argsort()
46 mindexer = s.argsort(kind="mergesort")
47 qindexer = s.argsort()
49 mexpected = np.argsort(s.values, kind="mergesort")
50 qexpected = np.argsort(s.values, kind="quicksort")
62 result = datetime_series.argsort()
|
| /dports/devel/py-pyopencl/pyopencl-2018.2/pyopencl/ |
| H A D | bitonic_sort.py | 108 argsort = 0
110 argsort = 1
113 argsort,
147 def get_program(self, letter, argsort, params): argument
151 NS="\\", argsort=argsort, inc=params[0], dir=params[1],
155 kid = Template(self.kernels_srcs[letter]).render(argsort=argsort)
161 def sort_b_prepare_wl(self, argsort, key_dtype, idx_dtype, shape, axis): argument
191 if argsort:
209 'BLO', argsort, (1, 1, key_ctype, idx_ctype, ds, ns))
232 prg = self.get_program(letter, argsort,
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/sort/ |
| H A D | sort.cc | 187 argsort<float, int32_t>(input, output, axis, is_ascend); in __anonb5140abd0202()
189 argsort<float, int64_t>(input, output, axis, is_ascend); in __anonb5140abd0202()
191 argsort<float, float>(input, output, axis, is_ascend); in __anonb5140abd0202()
193 argsort<float, double>(input, output, axis, is_ascend); in __anonb5140abd0202()
199 argsort<double, int32_t>(input, output, axis, is_ascend); in __anonb5140abd0202()
201 argsort<double, int64_t>(input, output, axis, is_ascend); in __anonb5140abd0202()
203 argsort<double, float>(input, output, axis, is_ascend); in __anonb5140abd0202()
205 argsort<double, double>(input, output, axis, is_ascend); in __anonb5140abd0202()
212 argsort<__fp16, __fp16>(input, output, axis, is_ascend); in __anonb5140abd0202()
223 argsort<int32_t, float>(input, output, axis, is_ascend); in __anonb5140abd0202()
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/runtime/contrib/sort/ |
| H A D | sort.cc | 190 argsort<float, int32_t>(input, output, axis, is_ascend); in __anone763a8020202()
192 argsort<float, int64_t>(input, output, axis, is_ascend); in __anone763a8020202()
194 argsort<float, float>(input, output, axis, is_ascend); in __anone763a8020202()
196 argsort<float, double>(input, output, axis, is_ascend); in __anone763a8020202()
202 argsort<double, int32_t>(input, output, axis, is_ascend); in __anone763a8020202()
204 argsort<double, int64_t>(input, output, axis, is_ascend); in __anone763a8020202()
206 argsort<double, float>(input, output, axis, is_ascend); in __anone763a8020202()
208 argsort<double, double>(input, output, axis, is_ascend); in __anone763a8020202()
215 argsort<__fp16, __fp16>(input, output, axis, is_ascend); in __anone763a8020202()
226 argsort<int32_t, float>(input, output, axis, is_ascend); in __anone763a8020202()
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/runtime/contrib/sort/ |
| H A D | sort.cc | 190 argsort<float, int32_t>(input, output, axis, is_ascend); in __anon9f6e80180202()
192 argsort<float, int64_t>(input, output, axis, is_ascend); in __anon9f6e80180202()
194 argsort<float, float>(input, output, axis, is_ascend); in __anon9f6e80180202()
196 argsort<float, double>(input, output, axis, is_ascend); in __anon9f6e80180202()
202 argsort<double, int32_t>(input, output, axis, is_ascend); in __anon9f6e80180202()
204 argsort<double, int64_t>(input, output, axis, is_ascend); in __anon9f6e80180202()
206 argsort<double, float>(input, output, axis, is_ascend); in __anon9f6e80180202()
208 argsort<double, double>(input, output, axis, is_ascend); in __anon9f6e80180202()
215 argsort<__fp16, __fp16>(input, output, axis, is_ascend); in __anon9f6e80180202()
226 argsort<int32_t, float>(input, output, axis, is_ascend); in __anon9f6e80180202()
[all …]
|
| /dports/devel/py-orange-canvas-core/orange-canvas-core-0.1.20/orangecanvas/canvas/items/tests/ |
| H A D | test_utils.py | 3 from ..utils import linspace, linspace_trunc, argsort, composition
43 argsort(seq), expected
46 argsort(seq, reverse=True), expected[::-1]
53 self.assertSequenceEqual(argsort(seq, key=sum), expected)
54 self.assertSequenceEqual(argsort(seq, key=sum, reverse=True),
|
| /dports/misc/py-gluoncv/gluon-cv-0.9.0/gluoncv/utils/viz/ |
| H A D | mask.py | 55 sorted_inds = np.argsort(-areas)
57 sorted_inds = np.argsort(-bboxes[:, 0])
59 sorted_inds = np.argsort(-bboxes[:, 1])
61 sorted_inds = np.argsort(-bboxes[:, 2])
63 sorted_inds = np.argsort(-bboxes[:, 3])
68 sorted_inds = np.argsort(range(len(masks)))
|
| /dports/math/py-numpoly/numpoly-1.2.3/numpoly/poly_function/ |
| H A D | sortable_proxy.py | 63 values = numpy.argsort(coefficients[idx][indices])
64 proxy[indices] = numpy.argsort(values)+numpy.max(proxy)+1
66 proxy = numpy.argsort(numpy.argsort(proxy.ravel())).reshape(proxy.shape)
|
| /dports/science/py-hiphive/hiphive-0.7.1/hiphive/structure_generation/ |
| H A D | phonon.py | 85 argsort = np.argsort(np.abs(w2_s))
86 w2_gamma = w2_s[argsort][:3]
87 e_sai = e_sai[argsort][3:]
88 w2_s = w2_s[argsort][3:]
|
| /dports/devel/py-pythran/pythran-0.11.0/pythran/pythonic/include/numpy/ |
| H A D | argsort.hpp | 12 types::ndarray<long, pS> argsort(types::ndarray<T, pS> const &a);
14 NUMPY_EXPR_TO_NDARRAY0_DECL(argsort);
16 DEFINE_FUNCTOR(pythonic::numpy, argsort);
|
| /dports/math/py-theano/Theano-1.0.5/theano/tensor/tests/ |
| H A D | test_sort.py | 13 from theano.tensor.sort import argsort, ArgSortOp
169 w = argsort(a)
172 gt = np.argsort(m_val)
178 w = argsort(a, axis)
182 gt = np.argsort(m_val, axis_val)
187 w2 = argsort(a)
190 gt = np.argsort(v_val)
196 l = argsort(a, axis, "mergesort")
200 gt = np.argsort(m_val, axis_val)
215 w2 = argsort(a, None)
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tests/python/relay/ |
| H A D | test_op_level6.py | 27 z = relay.argsort(x, axis=axis, is_ascend=is_ascend, dtype=dtype)
31 ref_res = np.argsort(x_data, axis=axis)
33 ref_res = np.argsort(-x_data, axis=axis)
56 np_indices = np.argsort(np_data, axis=axis)
58 np_indices = np.argsort(-np_data, axis=axis)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/relay/ |
| H A D | test_op_level6.py | 30 z = relay.argsort(x, axis=axis, is_ascend=is_ascend, dtype=dtype)
34 ref_res = np.argsort(x_data, axis=axis)
36 ref_res = np.argsort(-x_data, axis=axis)
61 np_indices = np.argsort(np_data, axis=axis)
63 np_indices = np.argsort(-np_data, axis=axis)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/relay/ |
| H A D | test_op_level6.py | 27 z = relay.argsort(x, axis=axis, is_ascend=is_ascend, dtype=dtype)
31 ref_res = np.argsort(x_data, axis=axis)
33 ref_res = np.argsort(-x_data, axis=axis)
56 np_indices = np.argsort(np_data, axis=axis)
58 np_indices = np.argsort(-np_data, axis=axis)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/topi/python/ |
| H A D | test_topi_sort.py | 27 "generic": (topi.argsort, topi.generic.schedule_argsort),
28 "gpu": (topi.cuda.argsort, topi.cuda.schedule_argsort),
47 np_indices = np.argsort(np_data, axis=axis)
49 np_indices = np.argsort(-np_data, axis=axis)
84 np_indices = np.argsort(np_data, axis=axis)
86 np_indices = np.argsort(-np_data, axis=axis)
|
| /dports/math/py-pandas/pandas-1.2.5/pandas/tests/arrays/categorical/ |
| H A D | test_sorting.py | 14 c.argsort(ascending=True), expected, check_dtype=False
19 c.argsort(ascending=False), expected, check_dtype=False
26 tm.assert_numpy_array_equal(np.argsort(c), expected, check_dtype=False)
29 np.argsort(c, kind="mergesort"), expected, check_dtype=False
34 np.argsort(c, axis=0)
38 np.argsort(c, order="C")
|
| /dports/math/py-pandas/pandas-1.2.5/pandas/tests/indexes/multi/ |
| H A D | test_sorting.py | 69 result = np.argsort(idx)
70 expected = idx.argsort()
83 np.argsort(idx, axis=1)
87 np.argsort(idx, kind="mergesort")
91 np.argsort(idx, order=("a", "b"))
271 result = idx.argsort()
272 expected = idx.values.argsort()
|
| /dports/math/py-statsmodels/statsmodels-0.13.1/statsmodels/sandbox/stats/ |
| H A D | stats_dhuard.py | 132 i = np.argsort(np.argsort(data)) + 1.
167 sortind = np.argsort(data)
169 self.ranking = np.argsort(sortind)
194 i = np.argsort(np.argsort(data)) + 1.
|
| /dports/math/py-statsmodels/statsmodels-0.13.1/statsmodels/examples/ |
| H A D | ex_univar_kde.py | 36 ix = np.argsort(support)
58 ix = np.argsort(support)
78 ix = np.argsort(support)
99 ix = np.argsort(support)
121 ix = np.argsort(support)
141 ix = np.argsort(support)
|
| /dports/math/py-statsmodels/statsmodels-0.13.1/statsmodels/stats/ |
| H A D | diagnostic_gen.py | 84 argsort = np.argsort(sort_var, kind=sort_method)
86 argsort = np.arange(observed.shape[0])
88 indices = np.array_split(argsort, bins, axis=0)
|
| /dports/science/py-ase/ase-3.22.0/ase/geometry/ |
| H A D | minkowski_reduction.py | 57 indices = np.argsort(np.linalg.norm(vs, axis=1))[:7]
90 H = H[np.argsort(norms, kind='merge')]
171 cell = cell[np.argsort(norms)[[1, 2, 0]]]
234 perm = np.argsort(pbc, kind='merge')[::-1] # stable sort
240 indices = np.argsort(norms)
247 invperm = np.argsort(perm)
|
| /dports/science/afni/afni-AFNI_21.3.16/src/pkundu/meica.libs/nibabel/nicom/ |
| H A D | dwiparams.py | 58 inds = np.argsort(w)[::-1]
97 inds = np.argsort(vals)[::-1]
122 scalers = scalers[np.argsort(inds)]
|