Searched defs:per_sample_weights_ (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qembeddingbag.cpp | 41 const std::optional<at::Tensor>& per_sample_weights_, in embedding_lookup_fallback_impl()
230 const std::optional<at::Tensor>& per_sample_weights_, in embedding_bag_nbit_impl()
402 const std::optional<at::Tensor>& per_sample_weights_, in embedding_bag_byte_impl()
563 const std::optional<at::Tensor>& per_sample_weights_, in embedding_bag_byte_helper()
661 const std::optional<at::Tensor>& per_sample_weights_, in _embedding_bag_nbit_helper()
765 const std::optional<at::Tensor>& per_sample_weights_, in embeddingbag_byte()
786 const std::optional<at::Tensor>& per_sample_weights_, in embeddingbag_4bit()
826 const std::optional<Tensor>& per_sample_weights_, in embedding_bag_byte_rowwise_offsets_out()
849 const std::optional<Tensor>& per_sample_weights_, in embedding_bag_4bit_rowwise_offsets_out()
884 const std::optional<Tensor>& per_sample_weights_, in embedding_bag_2bit_rowwise_offsets_out()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/ |
| H A D | EmbeddingBag.cu | 92 const PackedTensorAccessor32<float, 1, RestrictPtrTraits> per_sample_weights_, in embedding_bag_nbits_rowwise_offsets_kernel()
195 const std::optional<at::Tensor>& per_sample_weights_, in embedding_bag_byte_impl()
274 const std::optional<Tensor>& per_sample_weights_, in embedding_bag_byte_rowwise_offsets()
378 const std::optional<at::Tensor>& per_sample_weights_, in embedding_bag_4bit_impl()
456 const std::optional<Tensor>& per_sample_weights_, in embedding_bag_4bit_rowwise_offsets()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | embedding.h | 103 auto per_sample_weights_ = per_sample_weights; in embedding_bag() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | EmbeddingBag.cpp | 1512 const Tensor& per_sample_weights_, in _embedding_bag_dense_backward_cpu_sum_mean()
1602 const Tensor& per_sample_weights_ = *per_sample_weights__maybe_owned; in _embedding_bag_dense_backward_cpu() local
|