Lines Matching refs:inputData
50 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta, in softmaxSlowFloat32() argument
58 const float* inputBeg = inputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
82 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxFloat32() argument
90 tflite::optimized_ops::Softmax(param, convertShapeToTflshape(inputShape), inputData, 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
102 convertFloat16ToFloat32(inputData, &inputData_float32); in softmaxFloat16()
113 bool softmaxQuant8Impl(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8Impl() argument
133 const T* inputBeg = inputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
200 bool softmaxQuant8(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8() argument
225 return softmaxQuant8Impl(inputData, inputShape, beta, axis, inputMultiplier, inputLeftShift, in softmaxQuant8()