| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/ |
| H A D | qembedding_pack_test.py | 43 def init(self, num_embeddings, embedding_dim, op_func): argument
54 def init(self, num_embeddings, embedding_dim, batch_size, op_func): argument
68 def init(self, num_embeddings, embedding_dim, op_func): argument
78 def init(self, num_embeddings, embedding_dim, batch_size, op_func): argument
|
| H A D | embeddingbag_test.py | 55 def init(self, num_embeddings, embedding_dim, input_size, device): argument
|
| H A D | qatembedding_ops_test.py | 69 def init(self, num_embeddings, embedding_dim, input_size, device): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/ |
| H A D | embedding.cpp | 6 int64_t num_embeddings, in EmbeddingOptions()
11 int64_t num_embeddings, in EmbeddingBagOptions()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | embedding.h | 32 EmbeddingImpl(int64_t num_embeddings, int64_t embedding_dim) in EmbeddingImpl()
109 EmbeddingBagImpl(int64_t num_embeddings, int64_t embedding_dim) in EmbeddingBagImpl()
|
| /aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/configurations/ |
| H A D | model_configurations.json | 5 "num_embeddings": 1024, number
16 "num_embeddings": 1024, number
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/qat/modules/ |
| H A D | embedding_ops.py | 30 num_embeddings, argument
144 num_embeddings, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 15 …tepTensor(const Tensor& indices, const c10::SymInt& bdim_size, const c10::SymInt& num_embeddings) { in getStepTensor()
43 const auto num_embeddings = weight.size((*weight_bdim == 0) ? 1 : 0); in embedding_batch_rule() local
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 4548 …def _test_embedding_bag_unpack_fn(self, pack_fn, unpack_fn, num_embeddings, embedding_dim, bit_rat… argument
4567 self, bit_rate, num_embeddings, argument
4617 num_embeddings, embedding_dim, argument
4701 def test_embedding_bag_byte(self, num_embeddings, argument
4726 def test_embedding_bag_4bit(self, num_embeddings, argument
4753 def test_embedding_bag_2bit(self, num_embeddings, argument
4773 def test_embedding(self, num_embeddings, embedding_dim): argument
|
| H A D | test_quantized_module.py | 1232 def test_embedding_api(self, num_embeddings, embedding_dim, set_qconfig): argument
1271 def test_embedding_bag_api(self, num_embeddings, embedding_dim, num_offsets, set_qconfig): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/ops/ |
| H A D | test_embedding.py | 35 num_embeddings, argument
|
| H A D | test_embedding_bag.py | 35 num_embeddings, argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_embedding_ops.py | 51 num_embeddings, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | embedding_ops.py | 17 def __init__(self, num_embeddings, embedding_dim, dtype=torch.quint8): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 1116 def checkEmbeddingSerialization(self, qemb, num_embeddings, embedding_dim, indices, offsets, argument
2643 def __init__(self, num_embeddings, embedding_dim): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | EmbeddingBag.cpp | 1237 auto num_embeddings = weight.size(0); in embedding_bag() local
|