Searched defs:compressed_indices_mapping (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qembeddingbag.cpp | 42 const std::optional<at::Tensor>& compressed_indices_mapping, in embedding_lookup_fallback_impl()
231 const std::optional<at::Tensor>& compressed_indices_mapping, in embedding_bag_nbit_impl()
403 const std::optional<at::Tensor>& compressed_indices_mapping, in embedding_bag_byte_impl()
564 const std::optional<at::Tensor>& compressed_indices_mapping, in embedding_bag_byte_helper()
662 const std::optional<at::Tensor>& compressed_indices_mapping, in _embedding_bag_nbit_helper()
766 const std::optional<at::Tensor>& compressed_indices_mapping, in embeddingbag_byte()
787 const std::optional<at::Tensor>& compressed_indices_mapping, in embeddingbag_4bit()
827 const std::optional<Tensor>& compressed_indices_mapping, in embedding_bag_byte_rowwise_offsets_out()
850 const std::optional<Tensor>& compressed_indices_mapping, in embedding_bag_4bit_rowwise_offsets_out()
885 const std::optional<Tensor>& compressed_indices_mapping, in embedding_bag_2bit_rowwise_offsets_out()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/ |
| H A D | EmbeddingBag.cu | 93 const std::optional<Tensor>& compressed_indices_mapping, in embedding_bag_nbits_rowwise_offsets_kernel()
196 const std::optional<at::Tensor>& compressed_indices_mapping, in embedding_bag_byte_impl()
275 const std::optional<Tensor>& compressed_indices_mapping, in embedding_bag_byte_rowwise_offsets()
379 const std::optional<at::Tensor>& compressed_indices_mapping, in embedding_bag_4bit_impl()
457 const std::optional<Tensor>& compressed_indices_mapping, in embedding_bag_4bit_rowwise_offsets()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RowwisePrune.cpp | 33 auto compressed_indices_mapping = at::empty({mask.numel()}, in _rowwise_prune_helper() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | ops.cpp | 1137 const auto compressed_indices_mapping = in __anon11f46a8b3102() local
1175 const auto compressed_indices_mapping = in __anon11f46a8b3302() local
|