Searched refs:output_embedding_size (Results 1 – 4 of 4) sorted by relevance
65 int output_embedding_size) { in CheckQuantizationParams() argument66 if (bytes_per_embedding * 8 / quantization_bits < output_embedding_size) { in CheckQuantizationParams()
90 int num_buckets, int bytes_per_embedding, int output_embedding_size, in TFLiteEmbeddingExecutor() argument97 output_embedding_size_(output_embedding_size), in TFLiteEmbeddingExecutor()
26 int output_embedding_size);
117 int output_embedding_size, const TfLiteTensor* scales,