Lines Matching refs:outputShape
41 uint32_t outputHeight = getSizeOfDimension(outputShape, 1); \
42 uint32_t outputWidth = getSizeOfDimension(outputShape, 2); \
43 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); \
51 const Shape& outputShape) { in groupedConvFloat32() argument
109 const Shape& outputShape) { in groupedConvQuant8() argument
115 int32_t outputOffset = outputShape.offset; in groupedConvQuant8()
120 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in groupedConvQuant8()
127 CalculateActivationRange<T>(activation, outputShape, &output_activation_min, in groupedConvQuant8()
188 const Shape& outputShape);
197 const Shape& outputShape);
206 int32_t activation, T* outputData, const Shape& outputShape) { in groupedConvQuant8PerChannel() argument
211 int32_t outputOffset = outputShape.offset; in groupedConvQuant8PerChannel()
224 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in groupedConvQuant8PerChannel()
231 CalculateActivationRange<T>(activation, outputShape, &output_activation_min, in groupedConvQuant8PerChannel()
290 int32_t activation, _Float16* outputData, const Shape& outputShape) { in groupedConvFloat16() argument
296 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in groupedConvFloat16()
305 outputData_float32.data(), outputShape); in groupedConvFloat16()
316 int32_t activation, uint8_t* outputData, const Shape& outputShape);
323 int32_t activation, int8_t* outputData, const Shape& outputShape);