Lines Matching refs:outputData
51 int32_t axis, float* outputData, const Shape& /*outputShape*/) { in softmaxSlowFloat32() argument
60 float* outputBeg = outputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
83 float* outputData, const Shape& outputShape) { in softmaxFloat32() argument
91 convertShapeToTflshape(outputShape), outputData); in softmaxFloat32()
94 return softmaxSlowFloat32(inputData, inputShape, beta, axis, outputData, outputShape); in softmaxFloat32()
99 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16() argument
107 convertFloat32ToFloat16(outputData_float32, outputData); in softmaxFloat16()
115 T* outputData, const Shape& /*outputShape*/) { in softmaxQuant8Impl() argument
135 T* outputBeg = outputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
201 T* outputData, const Shape& outputShape) { in softmaxQuant8() argument
226 diffMin, outputData, outputShape); in softmaxQuant8()