Searched refs:token_embeddings (Results 1 – 1 of 1) sorted by relevance
650 std::vector<float> token_embeddings; in SetupModelInput() local665 if (!EmbedTokensPerMessage(tokens, &token_embeddings, &max_tokens)) { in SetupModelInput()735 model_->tflite_model_spec()->input_token_embeddings(), token_embeddings, in SetupModelInput()