42 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta,  in softmaxSlowFloat32()  argument
62                 sum += std::exp((*p - maxValue) * beta);  in softmaxSlowFloat32()
67                 *pOut = std::exp((*p - maxValue) * beta) / sum;  in softmaxSlowFloat32()
74 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis,  in softmaxFloat32()  argument
81         tflite::SoftmaxParams param = {.beta = beta};  in softmaxFloat32()
86         return softmaxSlowFloat32(inputData, inputShape, beta, axis, outputData, outputShape);  in softmaxFloat32()
90 bool softmaxFloat16(const _Float16* inputData, const Shape& inputShape, const float beta,  in softmaxFloat16()  argument
97     softmaxFloat32(inputData_float32.data(), inputShape, beta, axis, outputData_float32.data(),  in softmaxFloat16()
104 bool softmaxQuant8Impl(const uint8_t* inputData, const Shape& inputShape, const float beta,  in softmaxQuant8Impl()  argument
186 bool softmaxQuant8(const uint8_t* inputData, const Shape& inputShape, const float beta,  in softmaxQuant8()  argument
198             std::min(1.0 * beta * inputShape.scale * (1 << (31 - kScaledDiffIntegerBits)),  in softmaxQuant8()
211         tflite::SoftmaxParams param = {.beta = beta,  in softmaxQuant8()
219         return softmaxQuant8Impl(inputData, inputShape, beta, axis, inputMultiplier, inputLeftShift,  in softmaxQuant8()
256     float beta = (input.type == OperandType::TENSOR_FLOAT16)  in prepare()  local
260     NN_RET_CHECK_GT(beta, 0.0f);  in prepare()