Searched refs:ComputeLogits (Results 1 – 3 of 3) sorted by relevance
193 bool ComputeLogits(const Vector &input, Vector *scores) const;196 bool ComputeLogits(const VectorSpan<float> &input, Vector *scores) const;
234 bool EmbeddingNetwork::ComputeLogits(const VectorSpan<float> &input, in ComputeLogits() function in libtextclassifier::nlp_core::EmbeddingNetwork239 bool EmbeddingNetwork::ComputeLogits(const Vector &input, in ComputeLogits() function in libtextclassifier::nlp_core::EmbeddingNetwork303 return ComputeLogits(concat, scores); in ComputeFinalScores()
268 network.ComputeLogits(features, &scores); in InferInternal()398 selection_network_->ComputeLogits(features, &scores); in SuggestSelectionSymmetrical()