Home
last modified time | relevance | path

Searched defs:sorted_indices (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Drnn.cpp431 const Tensor& sorted_indices, in forward_helper()
521 auto sorted_indices = torch::Tensor(); in forward() local
536 const auto& sorted_indices = packed_input.sorted_indices(); in forward_with_packed_input() local
609 const Tensor& sorted_indices, in forward_helper()
671 auto sorted_indices = torch::Tensor(); in forward() local
687 const auto& sorted_indices = packed_input.sorted_indices(); in forward_with_packed_input() local
718 const Tensor& sorted_indices, in forward_helper()
768 auto sorted_indices = torch::Tensor(); in forward() local
783 const auto& sorted_indices = packed_input.sorted_indices(); in forward_with_packed_input() local
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/utils/
H A Drnn.h93 const Tensor& sorted_indices() const { in sorted_indices() function
126 Tensor sorted_indices = sorted_indices_.defined() in to() local
204 Tensor sorted_indices; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DLegacyThrustHelpers.cu22 …_kernel_thrust_helper(Tensor &linearIndex, Tensor &orig_indices, Tensor &sorted_indices, int64_t n… in index_put_with_sort_kernel_thrust_helper()
47 void embedding_dense_backward_cuda_scan(Tensor &sorted_indices, Tensor &count) { in embedding_dense_backward_cuda_scan()
89 int64_t embedding_backward_cuda_kernel_unique_by_key(const Tensor &sorted_indices, Tensor &segment_… in embedding_backward_cuda_kernel_unique_by_key()
H A DUniqueCub.cu66 const Tensor& sorted_indices, in compute_unique()
162 Tensor sorted_indices; in operator ()() local
249 Tensor sorted_indices; in operator ()() local
H A DIndexing.cu51 …const int64_t* sorted_indices, const int64_t* indices, const scalar_t* grad_output, scalar_t* grad… in indexing_backward_kernel()
129 …const int64_t* sorted_indices, const int64_t* indices, const scalar_t* grad_output, scalar_t* grad… in indexing_backward_kernel_stride_1()
185 …const int64_t* sorted_indices, const int64_t* indices, const scalar_t* grad_output, scalar_t* grad… in indexing_backward_kernel_small_stride()
229 …const int64_t* sorted_indices, const int64_t* indices, const float* grad_output, scalar_t* grad_we… in indexing_backward_kernel_quantized()
464 auto sorted_indices = at::empty_like(linearIndex, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in index_put_with_sort_kernel() local
594 auto sorted_indices = at::empty_like(linearIndex, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in index_put_with_sort_quantized() local
H A DUnique.cu44 const Tensor &sorted_indices, in compute_unique()
H A DEmbedding.cu295 auto sorted_indices = at::empty_like(indices, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in embedding_dense_backward_cuda() local
H A DEmbeddingBackwardKernel.cu215 const Tensor &sorted_indices, in embedding_backward_cuda_kernel()
H A DEmbeddingBag.cu186 auto sorted_indices = at::empty_like(indices, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in embedding_bag_backward_cuda_sum_avg() local
/aosp_15_r20/external/protobuf/src/google/protobuf/
H A Dgenerated_enum_util.cc70 int LookUpEnumName(const EnumEntry* enums, const int* sorted_indices, in LookUpEnumName()
84 const EnumEntry* enums, const int* sorted_indices, size_t size, in InitializeEnumStrings()
/aosp_15_r20/external/cronet/third_party/protobuf/src/google/protobuf/
H A Dgenerated_enum_util.cc70 int LookUpEnumName(const EnumEntry* enums, const int* sorted_indices, in LookUpEnumName()
84 const EnumEntry* enums, const int* sorted_indices, size_t size, in InitializeEnumStrings()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CPP/functions/
H A DCPPDetectionPostProcessLayer.cpp183 …std::vector<unsigned int> &sorted_indices, const unsigned int num_output, const unsigned int max_d… in SaveOutputs()
328 std::vector<unsigned int> sorted_indices; in run() local
/aosp_15_r20/external/libtextclassifier/native/utils/container/
H A Dbit-vector_test.cc55 std::vector<int32> sorted_indices = {3, 7}; in TEST() local
/aosp_15_r20/packages/providers/MediaProvider/jni/
DRedactionInfo.cpp163 std::vector<off64_t> sorted_indices; in getReadRanges() local
/aosp_15_r20/external/tensorflow/tensorflow/examples/speech_commands/
H A Dlabel_wav.cc89 Tensor sorted_indices(tensorflow::DT_INT32, {how_many_labels}); in GetTopLabels() local
/aosp_15_r20/external/ComputeLibrary/src/core/CPP/kernels/
H A DCPPNonMaximumSuppressionKernel.cpp114 std::vector<unsigned int> sorted_indices; in run() local
H A DCPPBoxWithNonMaximaSuppressionLimitKernel.cpp121 std::vector<int> NonMaximaSuppression(const ITensor *proposals, std::vector<int> sorted_indices, co… in NonMaximaSuppression()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DEmbedding.cpp193 auto sorted_indices = std::vector<index_t>(data_ptr, data_ptr + num_indices); in embedding_renorm_cpu_() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/sparse/
H A Dsparse_csr_matrix_ops.py299 def sorted_indices(self): member in SparseMatrix
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSoftMax.cu306 auto sorted_indices = at::empty({nnz}, indices.options()); in compute_pool_max() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dnon_max_suppression_op.cu.cc388 const int* sorted_indices = d_sorted_indices.flat<int>().data(); in DoNMS() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Ddetection_postprocess.cc458 std::vector<int> sorted_indices; in NonMaxSuppressionSingleClassHelper() local
/aosp_15_r20/bootable/libbootloader/gbl/libstorage/src/
H A Dgpt.rs240 let mut sorted_indices: [u8; GPT_MAX_NUM_ENTRIES] = from_fn(|i| i.try_into().unwrap()); in check_entries() localVariable
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse_csr.py3148 def sorted_indices(self): member in TestSparseCSR._construct_sp_matrix.FakeBscMatrix
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dir.py2592 def sorted_indices(arr): function