Home
last modified time | relevance | path

Searched refs:outputDepth (Results 1 – 5 of 5) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DTransposeConv2D.cpp147 for (uint32_t k = 0; k < outputDepth; k++) { in transposeConvNhwc()
167 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvNhwc()
173 for (uint32_t d = 0; d < outputDepth; d++, outPtr++) { in transposeConvNhwc()
235 for (uint32_t k = 0; k < outputDepth; k++) { in transposeConvNhwc()
258 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvNhwc()
342 std::vector<double> realMultiplier(outputDepth, 0.0); in transposeConvQuant8PerChannelNhwc()
343 std::vector<int32_t> outputMultiplier(outputDepth, 0); in transposeConvQuant8PerChannelNhwc()
344 std::vector<int32_t> outputShift(outputDepth, 0); in transposeConvQuant8PerChannelNhwc()
345 for (int i = 0; i < outputDepth; ++i) { in transposeConvQuant8PerChannelNhwc()
377 for (uint32_t k = 0; k < outputDepth; k++) { in transposeConvQuant8PerChannelNhwc()
[all …]
H A DReshape.cpp119 const int outputDepth = extOutputShape.Dims(3); in padGeneric() local
127 outputData, padValue, leftBPadding * outputHeight * outputWidth * outputDepth); in padGeneric()
133 leftHPadding * outputWidth * outputDepth); in padGeneric()
139 leftWPadding * outputDepth); in padGeneric()
158 outputDepth - rightDPadding), in padGeneric()
166 padValue, rightWPadding * outputDepth); in padGeneric()
173 padValue, rightHPadding * outputWidth * outputDepth); in padGeneric()
179 padValue, rightBPadding * outputHeight * outputWidth * outputDepth); in padGeneric()
H A DConv2D.cpp371 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc() local
376 auto realMultiplier = std::vector<double>(outputDepth, .0f); in convQuant8PerChannelNhwc()
377 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in convQuant8PerChannelNhwc()
378 auto outputShift = std::vector<int32_t>(outputDepth, .0f); in convQuant8PerChannelNhwc()
380 for (int i = 0; i < outputDepth; ++i) { in convQuant8PerChannelNhwc()
402 for (uint32_t d = 0; d < outputDepth; d++) { in convQuant8PerChannelNhwc()
436 outPtr += outputDepth; in convQuant8PerChannelNhwc()
464 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc() local
469 auto realMultiplier = std::vector<double>(outputDepth, .0f); in convQuant8PerChannelNhwc()
471 auto outputShift = std::vector<int32_t>(outputDepth, .0f); in convQuant8PerChannelNhwc()
[all …]
H A DGroupedConv2D.cpp43 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); \
44 uint32_t outputGroupDepth = outputDepth / numGroups;
213 auto realMultiplier = std::vector<double>(outputDepth, .0f); in groupedConvQuant8PerChannel()
214 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in groupedConvQuant8PerChannel()
215 auto outputShift = std::vector<int32_t>(outputDepth, 0); in groupedConvQuant8PerChannel()
217 for (int i = 0; i < outputDepth; ++i) { in groupedConvQuant8PerChannel()
H A DDepthwiseConv2D.cpp298 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in depthwiseConvQuant8PerChannelNhwc() local
303 auto realMultiplier = std::vector<double>(outputDepth, .0f); in depthwiseConvQuant8PerChannelNhwc()
304 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in depthwiseConvQuant8PerChannelNhwc()
305 auto outputShift = std::vector<int32_t>(outputDepth, .0f); in depthwiseConvQuant8PerChannelNhwc()
307 for (int i = 0; i < outputDepth; ++i) { in depthwiseConvQuant8PerChannelNhwc()