Lines Matching refs:outputShape

125     uint32_t outHeight = getSizeOfDimension(outputShape, 1);    \
126 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
137 const Shape& outputShape) { in depthwiseConvNhwc() argument
160 convertShapeToTflshape(outputShape), outputData); in depthwiseConvNhwc()
171 _Float16* outputData, const Shape& outputShape) { in depthwiseConvNhwc() argument
180 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in depthwiseConvNhwc()
185 outputShape); in depthwiseConvNhwc()
197 const Shape& outputShape) { in depthwiseConvNhwc() argument
208 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in depthwiseConvNhwc()
213 CalculateActivationRangeUint8(activation, outputShape, &output_activation_min, in depthwiseConvNhwc()
226 .output_offset = outputShape.offset, in depthwiseConvNhwc()
236 convertShapeToTflshape(outputShape), outputData); in depthwiseConvNhwc()
248 Shape outputShape) { in depthwiseConvNhwc() argument
259 std::vector<uint8_t> unsignedOutput(getNumberOfElements(outputShape)); in depthwiseConvNhwc()
260 outputShape.offset += 128; in depthwiseConvNhwc()
266 activation, unsignedOutput.data(), outputShape)); in depthwiseConvNhwc()
281 int32_t depthMultiplier, int32_t activation, T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannelNhwc() argument
294 uint32_t outputHeight = getSizeOfDimension(outputShape, 1); in depthwiseConvQuant8PerChannelNhwc()
295 uint32_t outputWidth = getSizeOfDimension(outputShape, 2); in depthwiseConvQuant8PerChannelNhwc()
296 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in depthwiseConvQuant8PerChannelNhwc()
299 int32_t outputOffset = outputShape.offset; in depthwiseConvQuant8PerChannelNhwc()
311 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in depthwiseConvQuant8PerChannelNhwc()
318 CalculateActivationRange<T>(activation, outputShape, &output_activation_min, in depthwiseConvQuant8PerChannelNhwc()
377 const Shape& outputShape) { in depthwiseConv() argument
381 NN_RET_CHECK(output.initialize(outputData, outputShape)); in depthwiseConv()
400 T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannel() argument
404 NN_RET_CHECK(output.initialize(outputData, outputShape)); in depthwiseConvQuant8PerChannel()