Searched refs:argsort (Results 1 – 25 of 26) sorted by relevance
12
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | topk_op_test.py | 109 indices = np.argsort(-inputs)[:k]
122 indices = np.argsort(-inputs, axis=1)
136 indices = np.argsort(-inputs, axis=1)[:, :k]
150 indices = np.argsort(-inputs, axis=1)[:, :k]
166 indices = np.argsort(-inputs, axis=1, kind="mergesort")[:, :k]
|
| D | eig_op_test.py | 97 perm = np.argsort(e.real + e.imag, -1)
104 perm = np.argsort(e.real + e.imag, -1)
233 idx = sort_ops.argsort(math_ops.real(e) + math_ops.imag(e), axis=-1)
|
| D | self_adjoint_eig_op_test.py | 100 perm = np.argsort(e, -1)
|
| D | lookup_ops_test.py | 1571 pairs = pairs[pairs[:, 0].argsort()]
|
| D | metrics_test.py | 1377 inds = np.argsort(-predictions)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sort_ops_test.py | 124 array_ops.gather(arr, sort_ops.argsort(arr)).eval())
132 np.argsort(arr, axis=axis),
133 sort_ops.argsort(arr, axis=axis).eval())
|
| D | sort_ops.py | 92 def argsort(values, axis=-1, direction='ASCENDING', stable=False, name=None): function
|
| D | image_ops_impl.py | 5450 sorted_scores_indices = sort_ops.argsort(
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sort_ops_test.py | 147 sorted_indices = np.argsort(inputs[0], axis=dimension)
189 sorted_indices = np.argsort(100 * inputs[0] + inputs[1])
262 indices = x.argsort()[::-1][:k]
298 indices = x.argsort(axis=1)[::, -1:-k - 1:-1]
362 indices = x.argsort(axis=1)[::, -1:-k - 1:-1]
|
| D | xla_ops_test.py | 426 shuffle_indices = np.argsort(np.random.randn(xs.shape[0]))
439 shuffle_indices = np.argsort(np.random.randn(xs.shape[1]))
451 xs = xs[np.argsort(np.random.randn(xs.shape[0]))]
452 xs = xs[:, np.argsort(np.random.randn(xs.shape[1]))]
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_math_ops_test.py | 171 self._testUnaryOp(np_math_ops.argsort, np.argsort, 'argsort')
176 np.testing.assert_equal(np_math_ops.argsort(a, kind='stable'), r)
|
| D | np_math_ops.py | 1163 def argsort(a, axis=-1, kind='quicksort', order=None): # pylint: disable=missing-docstring function
1178 return sort_ops.argsort(a, axis, stable=stable)
|
| /external/tensorflow/tensorflow/lite/examples/python/ |
| D | label_image.py | 93 top_k = results.argsort()[-5:][::-1]
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | label_wav_dir.py | 76 top_k = predictions.argsort()[-num_top_predictions:][::-1]
|
| D | label_wav.py | 68 top_k = predictions.argsort()[-num_top_predictions:][::-1]
|
| /external/tensorflow/tensorflow/examples/label_image/ |
| D | label_image.py | 135 top_k = results.argsort()[-5:][::-1]
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_permutation.py | 214 perm = sort_ops.argsort(perm, axis=-1)
|
| D | linear_operator_util.py | 474 inverse_perm = np.argsort(perm)
|
| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | imagenet_utils.py | 162 top_indices = pred.argsort()[-top:][::-1]
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | discretization.py | 128 merged = merged[merged[:, 0].argsort()]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_array_ops.py | 618 permutation = sort_ops.argsort(partitions, stable=True)
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | py_builtins.py | 618 order = sort_ops.argsort(mapped, direction=direction)
|
| /external/tensorflow/third_party/py/numpy/tf_numpy_api/ |
| D | tensorflow.experimental.numpy.pbtxt | 196 name: "argsort"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 464 name: "argsort"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 788 name: "argsort"
|
12