Lines Matching refs:outputShape
52 int32_t axis, float* outputData, const Shape& outputShape) { in softmaxSlowFloat32() argument
84 float* outputData, const Shape& outputShape) { in softmaxFloat32() argument
92 convertShapeToTflshape(outputShape), outputData); in softmaxFloat32()
95 return softmaxSlowFloat32(inputData, inputShape, beta, axis, outputData, outputShape); in softmaxFloat32()
100 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16() argument
104 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in softmaxFloat16()
107 outputShape); in softmaxFloat16()
116 T* outputData, const Shape& outputShape) { in softmaxQuant8Impl() argument
202 T* outputData, const Shape& outputShape) { in softmaxQuant8() argument
206 if ((inputShape.type == OperandType::TENSOR_QUANT8_ASYMM && outputShape.offset != 0) || in softmaxQuant8()
208 outputShape.offset != -128) || in softmaxQuant8()
209 outputShape.scale != 1.f / 256) { in softmaxQuant8()
227 diffMin, outputData, outputShape); in softmaxQuant8()