Lines Matching refs:outputShape
58 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32() argument
65 uint32_t batch_size = getSizeOfDimension(outputShape, 0); in fullyConnectedFloat32()
73 outputData, convertShapeToDims(outputShape)); in fullyConnectedFloat32()
80 outputData, convertShapeToDims(outputShape)); in fullyConnectedFloat32()
88 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16() argument
97 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in fullyConnectedFloat16()
100 outputDataFloat32.data(), outputShape); in fullyConnectedFloat16()
109 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8() argument
113 int32_t outputOffset = outputShape.offset; in fullyConnectedQuant8()
121 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, weightsShape, biasShape, outputShape, in fullyConnectedQuant8()
126 CalculateActivationRangeUint8(activation, outputShape, &outputActivationMin, in fullyConnectedQuant8()
142 convertShapeToDims(outputShape), &gemmContext); in fullyConnectedQuant8()
150 int8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8() argument
159 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, weightsShape, biasShape, outputShape, in fullyConnectedQuant8()
162 CalculateActivationRangeInt8(activation, outputShape, &outputActivationMin, in fullyConnectedQuant8()
168 params.output_offset = outputShape.offset; in fullyConnectedQuant8()
178 biasData, convertShapeToTflshape(outputShape), outputData); in fullyConnectedQuant8()