53 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta,  in softmaxSlowFloat32()  argument
56     const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis);  in softmaxSlowFloat32()
57     const uint32_t axisSize = getSizeOfDimension(inputShape, axis);  in softmaxSlowFloat32()
59             getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape));  in softmaxSlowFloat32()
85 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis,  in softmaxFloat32()  argument
87     int32_t ndim = getNumberOfDimensions(inputShape);  in softmaxFloat32()
88     NN_CHECK(handleNegativeAxis(inputShape, &axis));  in softmaxFloat32()
93         tflite::optimized_ops::Softmax(param, convertShapeToTflshape(inputShape), inputData,  in softmaxFloat32()
97         return softmaxSlowFloat32(inputData, inputShape, beta, axis, outputData, outputShape);  in softmaxFloat32()
101 bool softmaxFloat16(const _Float16* inputData, const Shape& inputShape, const float beta,  in softmaxFloat16()  argument
104     std::vector<float> inputData_float32(getNumberOfElements(inputShape));  in softmaxFloat16()
108     softmaxFloat32(inputData_float32.data(), inputShape, beta, axis, outputData_float32.data(),  in softmaxFloat16()
116 bool softmaxQuant8Impl(const T* inputData, const Shape& inputShape, const float beta, int32_t axis,  in softmaxQuant8Impl()  argument
131     const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis);  in softmaxQuant8Impl()
132     const uint32_t axisSize = getSizeOfDimension(inputShape, axis);  in softmaxQuant8Impl()
134             getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape));  in softmaxQuant8Impl()
203 bool softmaxQuant8(const T* inputData, const Shape& inputShape, const float beta, int32_t axis,  in softmaxQuant8()  argument
205     int32_t ndim = getNumberOfDimensions(inputShape);  in softmaxQuant8()
206     NN_CHECK(handleNegativeAxis(inputShape, &axis));  in softmaxQuant8()
208     if ((inputShape.type == OperandType::TENSOR_QUANT8_ASYMM && outputShape.offset != 0) ||  in softmaxQuant8()
209         (inputShape.type == OperandType::TENSOR_QUANT8_ASYMM_SIGNED &&  in softmaxQuant8()
218             std::min(1.0 * beta * inputShape.scale * (1 << (31 - kScaledDiffIntegerBits)),  in softmaxQuant8()
228     return softmaxQuant8Impl(inputData, inputShape, beta, axis, inputMultiplier, inputLeftShift,  in softmaxQuant8()