Lines Matching refs:outputData
43 int32_t axis, float* outputData, const Shape& outputShape) { in softmaxSlowFloat32() argument
52 float* outputBeg = outputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
75 float* outputData, const Shape& outputShape) { in softmaxFloat32() argument
83 convertShapeToTflshape(outputShape), outputData); in softmaxFloat32()
86 return softmaxSlowFloat32(inputData, inputShape, beta, axis, outputData, outputShape); in softmaxFloat32()
91 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16() argument
99 convertFloat32ToFloat16(outputData_float32, outputData); in softmaxFloat16()
106 uint8_t* outputData, const Shape& outputShape) { in softmaxQuant8Impl() argument
126 uint8_t* outputBeg = outputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
187 int32_t axis, uint8_t* outputData, const Shape& outputShape) { in softmaxQuant8() argument
216 convertShapeToTflshape(outputShape), outputData); in softmaxQuant8()
220 diffMin, outputData, outputShape); in softmaxQuant8()