Searched refs:scores_size (Results 1 – 2 of 2) sorted by relevance
76 std::vector<float> ComputeSoftmax(const float *scores, int scores_size) { in ComputeSoftmax() argument79 exp_scores.reserve(scores_size); in ComputeSoftmax()80 softmax.reserve(scores_size); in ComputeSoftmax()84 for (int i = 0; i < scores_size; ++i) { in ComputeSoftmax()89 for (int i = 0; i < scores_size; ++i) { in ComputeSoftmax()98 for (int i = 0; i < scores_size; ++i) { in ComputeSoftmax()
34 std::vector<float> ComputeSoftmax(const float *scores, int scores_size);