Lines Matching refs:bytes_per_embedding
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()
81 DequantizeAddNBit(scales, embeddings, bytes_per_embedding, in DequantizeAdd()