Searched refs:ComputeSoftmax (Results 1 – 6 of 6) sorted by relevance
72 std::vector<float> ComputeSoftmax(const std::vector<float> &scores) { in ComputeSoftmax() function73 return ComputeSoftmax(scores.data(), scores.size()); in ComputeSoftmax()76 std::vector<float> ComputeSoftmax(const float *scores, int scores_size) { in ComputeSoftmax() function
31 std::vector<float> ComputeSoftmax(const std::vector<float> &scores);34 std::vector<float> ComputeSoftmax(const float *scores, int scores_size);
34 std::vector<float> ComputeSoftmax(const std::vector<float> &scores,
73 std::vector<float> ComputeSoftmax(const std::vector<float> &scores, in ComputeSoftmax() function
169 const std::vector<float> softmax = ComputeSoftmax(scores); in FindLanguages()
1562 ComputeSoftmax(logits.data(), logits.dim(1)); in ModelClassifyText()3019 const std::vector<float> scores = ComputeSoftmax( in ModelClickContextScoreChunks()