50 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta,  in softmaxSlowFloat32()  argument
70                 sum += std::exp((*p - maxValue) * beta);  in softmaxSlowFloat32()
75                 *pOut = std::exp((*p - maxValue) * beta) / sum;  in softmaxSlowFloat32()
82 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis,  in softmaxFloat32()  argument
89         tflite::SoftmaxParams param = {.beta = beta};  in softmaxFloat32()
94         return softmaxSlowFloat32(inputData, inputShape, beta, axis, outputData, outputShape);  in softmaxFloat32()
98 bool softmaxFloat16(const _Float16* inputData, const Shape& inputShape, const float beta,  in softmaxFloat16()  argument
105     softmaxFloat32(inputData_float32.data(), inputShape, beta, axis, outputData_float32.data(),  in softmaxFloat16()
200 bool softmaxQuant8(const T* inputData, const Shape& inputShape, const float beta, int32_t axis,  in softmaxQuant8()  argument
215             std::min(1.0 * beta * inputShape.scale * (1 << (31 - kScaledDiffIntegerBits)),  in softmaxQuant8()
225     return softmaxQuant8Impl(inputData, inputShape, beta, axis, inputMultiplier, inputLeftShift,  in softmaxQuant8()
233     float beta = (input.type == OperandType::TENSOR_FLOAT16)  in prepare()  local
237     NN_RET_CHECK_GT(beta, 0.0f);  in prepare()