Lines Matching refs:outputShape
54 int32_t axis, float* outputData, const Shape& outputShape) { in softmaxSlowFloat32() argument
86 float* outputData, const Shape& outputShape) { in softmaxFloat32() argument
94 convertShapeToTflshape(outputShape), outputData); in softmaxFloat32()
97 return softmaxSlowFloat32(inputData, inputShape, beta, axis, outputData, outputShape); in softmaxFloat32()
102 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16() argument
106 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in softmaxFloat16()
109 outputShape); in softmaxFloat16()
118 T* outputData, const Shape& outputShape) { in softmaxQuant8Impl() argument
204 T* outputData, const Shape& outputShape) { in softmaxQuant8() argument
208 if ((inputShape.type == OperandType::TENSOR_QUANT8_ASYMM && outputShape.offset != 0) || in softmaxQuant8()
210 outputShape.offset != -128) || in softmaxQuant8()
211 outputShape.scale != 1.f / 256) { in softmaxQuant8()
229 diffMin, outputData, outputShape); in softmaxQuant8()