Searched defs:ComputeSoftmax (Results 1 – 4 of 4) sorted by relevance
72 std::vector<float> ComputeSoftmax(const std::vector<float> &scores) { in ComputeSoftmax() function76 std::vector<float> ComputeSoftmax(const float *scores, int scores_size) { in ComputeSoftmax() function
74 std::vector<float> ComputeSoftmax(const std::vector<float> &scores, in ComputeSoftmax() function
46 static void ComputeSoftmax(const Eigen::DefaultDevice& d, InT logits, in ComputeSoftmax() function
142 mlir::Value ComputeSoftmax(mlir::OpBuilder& builder, in ComputeSoftmax() function