Lines Matching refs:quantization_bits
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()
85 TC3_LOG(ERROR) << "Unsupported quantization_bits: " << quantization_bits; in DequantizeAdd()