Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DTransposeConv2D.cpp119 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); \
146 for (uint32_t k = 0; k < outputDepth; k++) { in transposeConvNhwc()
154 uint32_t outputIndex = hOutput * outputWidth * outputDepth + in transposeConvNhwc()
155 wOutput * outputDepth + k; in transposeConvNhwc()
166 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvNhwc()
172 for (uint32_t d = 0; d < outputDepth; d++, outPtr++) { in transposeConvNhwc()
234 for (uint32_t k = 0; k < outputDepth; k++) { in transposeConvNhwc()
242 uint32_t outputIndex = hOutput * outputWidth * outputDepth + in transposeConvNhwc()
243 wOutput * outputDepth + k; in transposeConvNhwc()
257 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvNhwc()
[all …]
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()
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()
DConv2D.cpp370 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc() local
375 auto realMultiplier = std::vector<double>(outputDepth, .0f); in convQuant8PerChannelNhwc()
376 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in convQuant8PerChannelNhwc()
377 auto outputShift = std::vector<int32_t>(outputDepth, .0f); in convQuant8PerChannelNhwc()
379 for (int i = 0; i < outputDepth; ++i) { in convQuant8PerChannelNhwc()
401 for (uint32_t d = 0; d < outputDepth; d++) { in convQuant8PerChannelNhwc()
435 outPtr += outputDepth; in convQuant8PerChannelNhwc()
463 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc() local
468 auto realMultiplier = std::vector<double>(outputDepth, .0f); in convQuant8PerChannelNhwc()
469 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in convQuant8PerChannelNhwc()
[all …]
DDepthwiseConv2D.cpp296 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in depthwiseConvQuant8PerChannelNhwc() local
301 auto realMultiplier = std::vector<double>(outputDepth, .0f); in depthwiseConvQuant8PerChannelNhwc()
302 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in depthwiseConvQuant8PerChannelNhwc()
303 auto outputShift = std::vector<int32_t>(outputDepth, .0f); in depthwiseConvQuant8PerChannelNhwc()
305 for (int i = 0; i < outputDepth; ++i) { in depthwiseConvQuant8PerChannelNhwc()