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()
26 int output_embedding_size);
91 int num_buckets, int bytes_per_embedding, int output_embedding_size, in TFLiteEmbeddingExecutor() argument99 output_embedding_size_(output_embedding_size), in TFLiteEmbeddingExecutor()
99 int num_buckets, int bytes_per_embedding, int output_embedding_size,