Lines Matching refs:outputShape

59                            float* outputData, const Shape& outputShape) {  in fullyConnectedFloat32()  argument
66 uint32_t batch_size = getSizeOfDimension(outputShape, 0); in fullyConnectedFloat32()
74 outputData, convertShapeToDims(outputShape)); in fullyConnectedFloat32()
81 outputData, convertShapeToDims(outputShape)); in fullyConnectedFloat32()
89 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16() argument
98 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in fullyConnectedFloat16()
101 outputDataFloat32.data(), outputShape); in fullyConnectedFloat16()
110 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8() argument
114 int32_t outputOffset = outputShape.offset; in fullyConnectedQuant8()
122 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, weightsShape, biasShape, outputShape, in fullyConnectedQuant8()
127 CalculateActivationRangeUint8(activation, outputShape, &outputActivationMin, in fullyConnectedQuant8()
143 convertShapeToDims(outputShape), &gemmContext); in fullyConnectedQuant8()
151 int8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8() argument
160 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, weightsShape, biasShape, outputShape, in fullyConnectedQuant8()
163 CalculateActivationRangeInt8(activation, outputShape, &outputActivationMin, in fullyConnectedQuant8()
169 params.output_offset = outputShape.offset; in fullyConnectedQuant8()
179 biasData, convertShapeToTflshape(outputShape), outputData); in fullyConnectedQuant8()