Searched defs:orig_indices (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | LegacyThrustHelpers.cu | 22 void index_put_with_sort_kernel_thrust_helper(Tensor &linearIndex, Tensor &orig_indices, Tensor &so… in index_put_with_sort_kernel_thrust_helper()
|
| D | Embedding.cu | 296 auto orig_indices = at::empty_like(indices, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in embedding_dense_backward_cuda() local
|
| D | EmbeddingBackwardKernel.cu | 214 const Tensor &orig_indices, in embedding_backward_cuda_kernel()
|
| D | EmbeddingBag.cu | 187 auto orig_indices = at::empty_like(indices, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in embedding_bag_backward_cuda_sum_avg() local
|
| D | Indexing.cu | 465 auto orig_indices = at::empty_like(linearIndex, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in index_put_with_sort_kernel() local
595 auto orig_indices = at::empty_like(linearIndex, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in index_put_with_sort_quantized() local
|
| /external/libtextclassifier/native/annotator/ |
| D | annotator_jni.cc | 381 const std::string& utf8_str, const std::pair<int, int>& orig_indices, in ConvertIndicesBMPUTF8() argument
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesScatterOps.cpp | 552 ArrayRef<std::optional<Tensor>> orig_indices, in maybe_permute_values()
|