Searched refs:token_embeddings (Results 1 – 2 of 2) sorted by relevance
153 token_embeddings = tf.nn.embedding_lookup(self._embeddings, current_id)154 lstm_outputs, current_state = self._lstm_cell(token_embeddings,
767 std::vector<float> token_embeddings; in SetupModelInput() local782 if (!EmbedTokensPerMessage(tokens, &token_embeddings, &max_tokens)) { in SetupModelInput()850 model_->tflite_model_spec()->input_token_embeddings(), token_embeddings, in SetupModelInput()