Lines Matching refs:outputShape
118 uint32_t outputHeight = getSizeOfDimension(outputShape, 1); \
119 uint32_t outputWidth = getSizeOfDimension(outputShape, 2); \
120 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); \
129 const Shape& outputShape) { in transposeConvNhwc() argument
136 memset(outputData, 0, getNumberOfElements(outputShape) * sizeof(float)); in transposeConvNhwc()
185 const TransposeConv2dParam& param, T* outputData, const Shape& outputShape) { in transposeConvNhwc() argument
191 uint32_t tempBufferByteSize = getNumberOfElements(outputShape) * sizeof(int32_t); in transposeConvNhwc()
205 int32_t outputOffset = outputShape.offset; in transposeConvNhwc()
210 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in transposeConvNhwc()
217 CalculateActivationRange<T>(activation, outputShape, &outputActivationMin, in transposeConvNhwc()
281 const Shape& outputShape) { in transposeConvNhwc() argument
286 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in transposeConvNhwc()
294 outputShape); in transposeConvNhwc()
304 const Shape& outputShape) { in transposeConv() argument
308 NN_RET_CHECK(output.initialize(outputData, outputShape)); in transposeConv()
321 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc() argument
327 uint32_t tempBufferByteSize = getNumberOfElements(outputShape) * sizeof(int32_t); in transposeConvQuant8PerChannelNhwc()
340 int32_t outputOffset = outputShape.offset; in transposeConvQuant8PerChannelNhwc()
352 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in transposeConvQuant8PerChannelNhwc()
359 CalculateActivationRange<T>(activation, outputShape, &outputActivationMin, in transposeConvQuant8PerChannelNhwc()
424 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel() argument
428 NN_RET_CHECK(output.initialize(outputData, outputShape)); in transposeConvQuant8PerChannel()