Lines Matching refs:outputShape

138     uint32_t outHeight = getSizeOfDimension(outputShape, 1);                      \
139 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
146 im2colDim.sizes[3] = (int)getSizeOfDimension(outputShape, 0); \
147 im2colDim.sizes[2] = (int)getSizeOfDimension(outputShape, 1); \
148 im2colDim.sizes[1] = (int)getSizeOfDimension(outputShape, 2); \
199 float* outputData, const Shape& outputShape) { in convNhwc() argument
219 convertShapeToDims(outputShape), need_im2colData ? im2colData : nullptr, im2colDim); in convNhwc()
228 uint8_t* outputData, const Shape& outputShape) { in convNhwc() argument
235 int32_t outputOffset = outputShape.offset; in convNhwc()
243 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in convNhwc()
248 CalculateActivationRangeUint8(activation, outputShape, &output_activation_min, in convNhwc()
270 convertShapeToDims(outputShape), in convNhwc()
282 int8_t* outputData, Shape outputShape) { in convNhwc() argument
293 std::vector<uint8_t> unsignedOutput(getNumberOfElements(outputShape)); in convNhwc()
294 outputShape.offset += 128; in convNhwc()
299 dilation_height_factor, activation, unsignedOutput.data(), outputShape)); in convNhwc()
311 _Float16* outputData, const Shape& outputShape) { in convNhwc() argument
317 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in convNhwc()
326 dilation_height_factor, activation, outputData_float32.data(), outputShape); in convNhwc()
338 const Shape& outputShape) { in conv() argument
342 NN_RET_CHECK(output.initialize(outputData, outputShape)); in conv()
359 const Shape& outputShape) { in convQuant8PerChannelNhwc() argument
369 uint32_t outputHeight = getSizeOfDimension(outputShape, 1); in convQuant8PerChannelNhwc()
370 uint32_t outputWidth = getSizeOfDimension(outputShape, 2); in convQuant8PerChannelNhwc()
371 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc()
374 int32_t outputOffset = outputShape.offset; in convQuant8PerChannelNhwc()
386 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in convQuant8PerChannelNhwc()
393 CalculateActivationRangeUint8(activation, outputShape, &output_activation_min, in convQuant8PerChannelNhwc()
452 const Shape& outputShape) { in convQuant8PerChannelNhwc() argument
462 uint32_t outputHeight = getSizeOfDimension(outputShape, 1); in convQuant8PerChannelNhwc()
463 uint32_t outputWidth = getSizeOfDimension(outputShape, 2); in convQuant8PerChannelNhwc()
464 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc()
467 int32_t outputOffset = outputShape.offset; in convQuant8PerChannelNhwc()
479 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in convQuant8PerChannelNhwc()
484 CalculateActivationRangeInt8(activation, outputShape, &output_activation_min, in convQuant8PerChannelNhwc()
489 convParams.output_offset = outputShape.offset; in convQuant8PerChannelNhwc()
504 convertShapeToTflshape(outputShape), outputData); in convQuant8PerChannelNhwc()
515 T* outputData, const Shape& outputShape) { in convQuant8PerChannel() argument
519 NN_RET_CHECK(output.initialize(outputData, outputShape)); in convQuant8PerChannel()