Lines Matching refs:outputShape
127 uint32_t outHeight = getSizeOfDimension(outputShape, 1); \
128 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
139 const Shape& outputShape) { in depthwiseConvNhwc() argument
162 convertShapeToTflshape(outputShape), outputData); in depthwiseConvNhwc()
173 _Float16* outputData, const Shape& outputShape) { in depthwiseConvNhwc() argument
182 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in depthwiseConvNhwc()
187 outputShape); in depthwiseConvNhwc()
199 const Shape& outputShape) { in depthwiseConvNhwc() argument
210 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in depthwiseConvNhwc()
215 CalculateActivationRangeUint8(activation, outputShape, &output_activation_min, in depthwiseConvNhwc()
228 .output_offset = outputShape.offset, in depthwiseConvNhwc()
238 convertShapeToTflshape(outputShape), outputData); in depthwiseConvNhwc()
250 Shape outputShape) { in depthwiseConvNhwc() argument
261 std::vector<uint8_t> unsignedOutput(getNumberOfElements(outputShape)); in depthwiseConvNhwc()
262 outputShape.offset += 128; in depthwiseConvNhwc()
268 activation, unsignedOutput.data(), outputShape)); in depthwiseConvNhwc()
283 int32_t depthMultiplier, int32_t activation, T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannelNhwc() argument
296 uint32_t outputHeight = getSizeOfDimension(outputShape, 1); in depthwiseConvQuant8PerChannelNhwc()
297 uint32_t outputWidth = getSizeOfDimension(outputShape, 2); in depthwiseConvQuant8PerChannelNhwc()
298 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in depthwiseConvQuant8PerChannelNhwc()
301 int32_t outputOffset = outputShape.offset; in depthwiseConvQuant8PerChannelNhwc()
313 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in depthwiseConvQuant8PerChannelNhwc()
320 CalculateActivationRange<T>(activation, outputShape, &output_activation_min, in depthwiseConvQuant8PerChannelNhwc()
379 const Shape& outputShape) { in depthwiseConv() argument
383 NN_RET_CHECK(output.initialize(outputData, outputShape)); in depthwiseConv()
402 T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannel() argument
406 NN_RET_CHECK(output.initialize(outputData, outputShape)); in depthwiseConvQuant8PerChannel()