Lines Matching refs:inputData
53 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta, in softmaxSlowFloat32() argument
61 const float* inputBeg = inputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
85 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxFloat32() argument
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
105 convertFloat16ToFloat32(inputData, &inputData_float32); in softmaxFloat16()
116 bool softmaxQuant8Impl(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8Impl() argument
136 const T* inputBeg = inputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
203 bool softmaxQuant8(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8() argument
228 return softmaxQuant8Impl(inputData, inputShape, beta, axis, inputMultiplier, inputLeftShift, in softmaxQuant8()