vendor: OpenCV 5.0.0 snapshot at 40738fb16ceddb5fb3fea747585f7ce6abb0605b

2026-08-22 00:10:33 +08:00
commit f7f077da11
6933 changed files with 2335208 additions and 0 deletions
@@ -0,0 +1,129 @@
+/*++
+
+Copyright (c) Microsoft Corporation. All rights reserved.
+
+Licensed under the MIT License.
+
+Module Name:
+
+    softmax.h
+
+Abstract:
+
+    This module includes kernel function prototypes and helper functions for
+    softmax.
+
+--*/
+
+#pragma once
+
+#include "mlasi.h"
+
+struct MLAS_SOFTMAX_DISPATCH {
+    /**
+     * @brief Compute the hyperbolic tangent function for each element of the input array
+     * @param Input         Address of the input array. Valid in [-3.51562, 3.51562].
+     * @param Output        Address of the output array. Could be the same as the input array.
+     * @param N             Number of elements in the input array
+     */
+    typedef void(Tanh_Fp16_Fn)(
+        const MLAS_FP16* Input,
+        MLAS_FP16* Output,
+        size_t N
+    );
+
+    Tanh_Fp16_Fn* Tanh_Fp16 = nullptr;
+
+    /**
+     * @brief Compute the softcap function for each element of the input array. Use tanh activation.
+     * @param Input         Address of the input array. Valid if input / softcap in [-3.51562, 3.51562].
+     * @param Output        Address of the output array. Could be the same as the input array.
+     * @param N             Number of elements in the input array
+     * @param Softcap       The softcap value
+     */
+    typedef void(Softcap_Fp16_Fn)(
+        const MLAS_FP16* Input,
+        MLAS_FP16* Output,
+        size_t N,
+        const MLAS_FP16 Softcap
+    );
+
+    Softcap_Fp16_Fn* Softcap_Fp16 = nullptr;
+
+    /**
+     * @brief Compute the exponential function for each element of the input array.
+     * @param Input         Address of the input array. Valid in [-17.3287, 11.0904].
+     * @param Output        Address of the output array. Could be the same as the input array.
+     * @param N             Number of elements in the input array
+     */
+    typedef void(Exp_Fp16_Fn)(
+        const MLAS_FP16* Input,
+        MLAS_FP16* Output,
+        size_t N
+    );
+
+    Exp_Fp16_Fn* Exp_Fp16 = nullptr;
+
+    /**
+     * @brief Find the max value among the input array
+     * @param Input         Address of the input array
+     * @param N             Number of elements in the input array
+     */
+    typedef MLAS_FP16(ReduceMax_Fp16_Fn)(
+        const MLAS_FP16* Input,
+        size_t N
+    );
+
+    ReduceMax_Fp16_Fn* ReduceMax_Fp16 = nullptr;
+
+    /**
+     * @brief Compute the expotential function for each element of the input array and returnt he sum. It has smaller
+     *        dynamic range for the input than Exp_Fp16_Fn thus is faster.
+     * @param Input         Address of the input array. Valid in [-10.7438, 10.7438]
+     * @param Output        Address of the output array. Could be the same as the input array or nullptr.
+     * @param N             Number of elements in the input array
+     * @param NegativeMaximum   The negative of the maximum value in the input array
+     */
+    typedef MLAS_FP16(SumExp_Fp16_Fn)(
+        const MLAS_FP16* Input,
+        MLAS_FP16* Output,
+        size_t N,
+        const MLAS_FP16 NegativeMaximum
+    );
+
+    SumExp_Fp16_Fn* SumExp_Fp16 = nullptr;
+
+    /**
+     * @brief Compute the softmax output for each element of the input array. input / sum.
+     * @param Input         Address of the input array. Values of exp(x)
+     * @param Output        Address of the output array. Could be the same as the input array.
+     * @param N             Number of elements in the input array
+     * @param Sum           Sum of exp(input)
+     */
+    typedef void(Softmax_Fp16_Fn)(
+        const MLAS_FP16* Input,
+        MLAS_FP16* Output,
+        size_t N,
+        const MLAS_FP16 Sum
+    );
+
+    Softmax_Fp16_Fn* Softmax_Fp16 = nullptr;
+
+    /**
+     * @brief Compute the log softmax output for each element of the input array. input - max - logSum
+     * @param Input         Address of the input array
+     * @param Output        Address of the output array. Could be the same as the input array.
+     * @param N             Number of elements in the input array
+     * @param NagativeMaximum   The negative of the maximum value in the input array
+     * @param LogSum        The logarithm of the sum of the exponential function of the input array
+     */
+    typedef void(LogSoftmax_Fp16_Fn)(
+        const MLAS_FP16* Input,
+        MLAS_FP16* Output,
+        size_t N,
+        const MLAS_FP16 NagativeMaximum,
+        const MLAS_FP16 LogSum
+    );
+
+    LogSoftmax_Fp16_Fn* LogSoftmax_Fp16 = nullptr;
+};