Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DTransposeConv2D.cpp114 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); \
143 for (uint32_t k = 0; k < outputDepth; k++) { in transposeConvNhwc()
151 uint32_t outputIndex = hOutput * outputWidth * outputDepth + in transposeConvNhwc()
152 wOutput * outputDepth + k; in transposeConvNhwc()
163 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvNhwc()
169 for (uint32_t d = 0; d < outputDepth; d++, outPtr++) { in transposeConvNhwc()
231 for (uint32_t k = 0; k < outputDepth; k++) { in transposeConvNhwc()
239 uint32_t outputIndex = hOutput * outputWidth * outputDepth + in transposeConvNhwc()
240 wOutput * outputDepth + k; in transposeConvNhwc()
254 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvNhwc()
[all …]
DReshape.cpp125 const int outputDepth = extOutputShape.Dims(3); in padGeneric() local
133 outputData, padValue, leftBPadding * outputHeight * outputWidth * outputDepth); in padGeneric()
139 leftHPadding * outputWidth * outputDepth); in padGeneric()
145 leftWPadding * outputDepth); in padGeneric()
164 outputDepth - rightDPadding), in padGeneric()
172 padValue, rightWPadding * outputDepth); in padGeneric()
179 padValue, rightHPadding * outputWidth * outputDepth); in padGeneric()
185 padValue, rightBPadding * outputHeight * outputWidth * outputDepth); in padGeneric()
DConv2D.cpp368 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc() local
373 auto realMultiplier = std::vector<double>(outputDepth, .0f); in convQuant8PerChannelNhwc()
374 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in convQuant8PerChannelNhwc()
375 auto outputShift = std::vector<int32_t>(outputDepth, .0f); in convQuant8PerChannelNhwc()
377 for (uint32_t i = 0; i < outputDepth; ++i) { in convQuant8PerChannelNhwc()
399 for (uint32_t d = 0; d < outputDepth; d++) { in convQuant8PerChannelNhwc()
433 outPtr += outputDepth; in convQuant8PerChannelNhwc()
461 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc() local
466 auto realMultiplier = std::vector<double>(outputDepth, .0f); in convQuant8PerChannelNhwc()
467 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in convQuant8PerChannelNhwc()
[all …]
DGroupedConv2D.cpp47 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); \
48 uint32_t outputGroupDepth = outputDepth / numGroups;
218 auto realMultiplier = std::vector<double>(outputDepth, .0f); in groupedConvQuant8PerChannel()
219 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in groupedConvQuant8PerChannel()
220 auto outputShift = std::vector<int32_t>(outputDepth, 0); in groupedConvQuant8PerChannel()
222 for (uint32_t i = 0; i < outputDepth; ++i) { in groupedConvQuant8PerChannel()
DDepthwiseConv2D.cpp294 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in depthwiseConvQuant8PerChannelNhwc() local
299 auto realMultiplier = std::vector<double>(outputDepth, .0f); in depthwiseConvQuant8PerChannelNhwc()
300 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in depthwiseConvQuant8PerChannelNhwc()
301 auto outputShift = std::vector<int32_t>(outputDepth, .0f); in depthwiseConvQuant8PerChannelNhwc()
303 for (uint32_t i = 0; i < outputDepth; ++i) { in depthwiseConvQuant8PerChannelNhwc()