Searched refs:bytes_per_embedding (Results 1 – 5 of 5) sorted by relevance
| /external/libtextclassifier/annotator/ |
| D | quantization.cc | 29 int bytes_per_embedding, const int num_sparse_features, in DequantizeAdd8bit() argument
36 embeddings[bucket_id * bytes_per_embedding + k]); in DequantizeAdd8bit()
41 int bytes_per_embedding, int num_sparse_features, in DequantizeAddNBit() argument
50 uint16 data = embeddings[bucket_id * bytes_per_embedding + read16_offset]; in DequantizeAddNBit()
53 if (read16_offset < bytes_per_embedding - 1) { in DequantizeAddNBit()
54 data |= embeddings[bucket_id * bytes_per_embedding + read16_offset + 1] in DequantizeAddNBit()
64 bool CheckQuantizationParams(int bytes_per_embedding, int quantization_bits, in CheckQuantizationParams() argument
66 if (bytes_per_embedding * 8 / quantization_bits < output_embedding_size) { in CheckQuantizationParams()
74 int bytes_per_embedding, int num_sparse_features, in DequantizeAdd() argument
78 DequantizeAdd8bit(scales, embeddings, bytes_per_embedding, in DequantizeAdd()
[all …]
|
| D | quantization_test.cc | 46 const int bytes_per_embedding = 4; in TEST() local
51 DequantizeAdd(scales.data(), embeddings.data(), bytes_per_embedding, in TEST()
69 DequantizeAdd(scales.data(), embeddings.data(), bytes_per_embedding, in TEST()
86 const int bytes_per_embedding = 4; in TEST() local
93 std::vector<uint8> embeddings(bytes_per_embedding * num_buckets); in TEST()
98 DequantizeAdd(scales.data(), embeddings.data(), bytes_per_embedding, in TEST()
109 const int bytes_per_embedding = 4; in TEST() local
116 std::vector<uint8> embeddings(bytes_per_embedding * num_buckets, 0xFF); in TEST()
119 DequantizeAdd(scales.data(), embeddings.data(), bytes_per_embedding, in TEST()
129 const int bytes_per_embedding = 4; in TEST() local
[all …]
|
| D | model-executor.cc | 76 int bytes_per_embedding = embeddings->dims->data[1]; in FromBuffer() local
77 if (!CheckQuantizationParams(bytes_per_embedding, quantization_bits, in FromBuffer()
84 std::move(executor), quantization_bits, num_buckets, bytes_per_embedding, in FromBuffer()
91 int num_buckets, int bytes_per_embedding, int output_embedding_size, in TFLiteEmbeddingExecutor() argument
98 bytes_per_embedding_(bytes_per_embedding), in TFLiteEmbeddingExecutor()
|
| D | quantization.h | 25 bool CheckQuantizationParams(int bytes_per_embedding, int quantization_bits,
33 int bytes_per_embedding, int num_sparse_features,
|
| D | model-executor.h | 99 int num_buckets, int bytes_per_embedding, int output_embedding_size,
|