Lines Matching refs:outputShape

117     uint32_t outputHeight = getSizeOfDimension(outputShape, 1);                 \
118 uint32_t outputWidth = getSizeOfDimension(outputShape, 2); \
119 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); \
128 const Shape& outputShape) { in transposeConvNhwc() argument
135 memset(outputData, 0, getNumberOfElements(outputShape) * sizeof(float)); in transposeConvNhwc()
184 const TransposeConv2dParam& param, T* outputData, const Shape& outputShape) { in transposeConvNhwc() argument
190 uint32_t tempBufferByteSize = getNumberOfElements(outputShape) * sizeof(int32_t); in transposeConvNhwc()
204 int32_t outputOffset = outputShape.offset; in transposeConvNhwc()
209 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in transposeConvNhwc()
216 CalculateActivationRange<T>(activation, outputShape, &outputActivationMin, in transposeConvNhwc()
280 const Shape& outputShape) { in transposeConvNhwc() argument
285 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in transposeConvNhwc()
293 outputShape); in transposeConvNhwc()
303 const Shape& outputShape) { in transposeConv() argument
307 NN_RET_CHECK(output.initialize(outputData, outputShape)); in transposeConv()
320 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc() argument
326 uint32_t tempBufferByteSize = getNumberOfElements(outputShape) * sizeof(int32_t); in transposeConvQuant8PerChannelNhwc()
339 int32_t outputOffset = outputShape.offset; in transposeConvQuant8PerChannelNhwc()
351 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in transposeConvQuant8PerChannelNhwc()
358 CalculateActivationRange<T>(activation, outputShape, &outputActivationMin, in transposeConvQuant8PerChannelNhwc()
423 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel() argument
427 NN_RET_CHECK(output.initialize(outputData, outputShape)); in transposeConvQuant8PerChannel()