Searched refs:softmax (Results 1 – 5 of 5) sorted by relevance
74 std::vector<float> softmax; in ComputeSoftmax() local77 softmax.reserve(scores.size()); in ComputeSoftmax()94 softmax.push_back(exp_scores[i] / denominator); in ComputeSoftmax()96 return softmax; in ComputeSoftmax()
185 return proto_->has_softmax() ? proto_->softmax().rows() : 0; in softmax_num_rows()190 return proto_->has_softmax() ? proto_->softmax().cols() : 0; in softmax_num_cols()195 return proto_->has_softmax() ? proto_->softmax().value().data() : nullptr; in softmax_weights()
363 const EmbeddingNetworkParams::Matrix softmax = model->GetSoftmaxMatrix(); in EmbeddingNetwork() local365 if (!InitNonQuantizedMatrix(softmax, &softmax_weights_) || in EmbeddingNetwork()
67 optional MatrixParams softmax = 4; field
160 return trimmed_proto_.softmax().rows(); in softmax_num_rows()165 return trimmed_proto_.softmax().cols(); in softmax_num_cols()