Searched refs:quantization_bits (Results 1 – 5 of 5) sorted by relevance
| /external/libtextclassifier/ |
| 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 | 43 int quantization_bits) { in Instance() argument
77 if (!CheckQuantizationParams(bytes_per_embedding, quantization_bits, in Instance()
84 std::move(model), quantization_bits, num_buckets, bytes_per_embedding, in Instance()
89 std::unique_ptr<const tflite::FlatBufferModel> model, int quantization_bits, in TFLiteEmbeddingExecutor() argument
94 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 | 108 int quantization_bits);
116 int quantization_bits, int num_buckets, int bytes_per_embedding,
|