Searched refs:quantization_bits (Results 1 – 5 of 5) sorted by relevance
| /external/libtextclassifier/native/annotator/ |
| D | quantization.cc | 42 int quantization_bits, int bucket_id, float* dest, in DequantizeAddNBit() argument
44 const int quantization_bias = 1 << (quantization_bits - 1); in DequantizeAddNBit()
47 const int bit_offset = i * quantization_bits; in DequantizeAddNBit()
57 int value = (data >> (bit_offset % 8)) & ((1 << quantization_bits) - 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()
75 int quantization_bits, int bucket_id, float* dest, in DequantizeAdd() argument
77 if (quantization_bits == 8) { in DequantizeAdd()
80 } else if (quantization_bits != 8) { in DequantizeAdd()
82 num_sparse_features, quantization_bits, bucket_id, dest, in DequantizeAdd()
[all …]
|
| D | quantization_test.cc | 45 const int quantization_bits = 8; in TEST() local
52 num_sparse_features, quantization_bits, bucket_id, in TEST()
70 num_sparse_features, quantization_bits, bucket_id, in TEST()
89 const int quantization_bits = 1; in TEST() local
99 num_sparse_features, quantization_bits, bucket_id, dest.data(), in TEST()
112 const int quantization_bits = 1; in TEST() local
120 num_sparse_features, quantization_bits, bucket_id, dest.data(), in TEST()
132 const int quantization_bits = 3; in TEST() local
145 num_sparse_features, quantization_bits, bucket_id, dest.data(), in TEST()
|
| D | model-executor.cc | 47 int quantization_bits, in FromBuffer() argument
77 if (!CheckQuantizationParams(bytes_per_embedding, quantization_bits, in FromBuffer()
84 std::move(executor), quantization_bits, num_buckets, bytes_per_embedding, in FromBuffer()
90 std::unique_ptr<TfLiteModelExecutor> executor, int quantization_bits, in TFLiteEmbeddingExecutor() argument
96 quantization_bits_(quantization_bits), in TFLiteEmbeddingExecutor()
|
| D | quantization.h | 25 bool CheckQuantizationParams(int bytes_per_embedding, int quantization_bits,
34 int quantization_bits, int bucket_id, float* dest,
|
| D | model-executor.h | 81 int quantization_bits,
98 std::unique_ptr<TfLiteModelExecutor> executor, int quantization_bits,
|