Searched defs:argsort (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseBinaryOpIntersectionCommon.h | 302 const auto argsort = nnz_arange.narrow(-1, 0, probably_coalesced._nnz()); variable
309 Tensor sorted, argsort; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | sort_ops.py | 88 def argsort(values, axis=-1, direction='ASCENDING', stable=False, name=None): function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Sorting.cpp | 983 Tensor argsort(const Tensor & self, int64_t dim, bool descending) { in argsort() function
987 Tensor argsort(const Tensor & self, bool stable, int64_t dim, bool descending) { in argsort() function
|
| H A D | TensorCompare.cpp | 795 Tensor argsort(const Tensor& /*self*/, Dimname /*dim*/, bool /*keepdim*/) { in argsort() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | SparseBinaryOpIntersectionKernel.cu | 80 const Tensor& argsort, in binary_op_intersection_kernel()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset11.py | 570 def argsort(g: jit_utils.GraphContext, self, dim, decending, out=None): function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | utils.py | 842 def argsort(seq) -> List[int]: function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| H A D | np_math_ops.py | 1169 def argsort(a, axis=-1, kind='quicksort', order=None): # pylint: disable=missing-docstring function
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _funcs_impl.py | 1399 def argsort(a: ArrayLike, axis=-1, kind=None, order: NotImplementedType = None): function
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 3091 std::vector<std::size_t> argsort(sizes.size()); in _maybe_overlapping_memory() local
|