Searched refs:outputDepth (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | TransposeConv2D.cpp | 109 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); \
136 for (uint32_t k = 0; k < outputDepth; k++) { in transposeConvNhwc()
144 uint32_t outputIndex = hOutput * outputWidth * outputDepth + in transposeConvNhwc()
145 wOutput * outputDepth + k; in transposeConvNhwc()
156 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvNhwc()
162 for (uint32_t d = 0; d < outputDepth; d++, outPtr++) { in transposeConvNhwc()
224 for (uint32_t k = 0; k < outputDepth; k++) { in transposeConvNhwc()
232 uint32_t outputIndex = hOutput * outputWidth * outputDepth + in transposeConvNhwc()
233 wOutput * outputDepth + k; in transposeConvNhwc()
247 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvNhwc()
[all …]
|
| D | Reshape.cpp | 111 const int outputDepth = extOutputShape.Dims(3); in padGeneric() local
119 outputData, padValue, leftBPadding * outputHeight * outputWidth * outputDepth); in padGeneric()
125 leftHPadding * outputWidth * outputDepth); in padGeneric()
131 leftWPadding * outputDepth); in padGeneric()
150 outputDepth - rightDPadding), in padGeneric()
158 padValue, rightWPadding * outputDepth); in padGeneric()
165 padValue, rightHPadding * outputWidth * outputDepth); in padGeneric()
171 padValue, rightBPadding * outputHeight * outputWidth * outputDepth); in padGeneric()
|
| D | GroupedConv2D.cpp | 39 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); \
40 uint32_t outputGroupDepth = outputDepth / numGroups;
190 auto realMultiplier = std::vector<double>(outputDepth, .0f); in groupedConvQuant8PerChannel()
191 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in groupedConvQuant8PerChannel()
192 auto outputShift = std::vector<int32_t>(outputDepth, 0); in groupedConvQuant8PerChannel()
194 for (int i = 0; i < outputDepth; ++i) { in groupedConvQuant8PerChannel()
|
| D | DepthwiseConv2D.cpp | 170 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in depthwiseConvQuant8PerChannel() local
175 auto realMultiplier = std::vector<double>(outputDepth, .0f); in depthwiseConvQuant8PerChannel()
176 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in depthwiseConvQuant8PerChannel()
177 auto outputShift = std::vector<int32_t>(outputDepth, .0f); in depthwiseConvQuant8PerChannel()
179 for (int i = 0; i < outputDepth; ++i) { in depthwiseConvQuant8PerChannel()
|
| D | Conv2D.cpp | 301 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc() local
306 auto realMultiplier = std::vector<double>(outputDepth, .0f); in convQuant8PerChannelNhwc()
307 auto outputMultiplier = std::vector<int32_t>(outputDepth, 0); in convQuant8PerChannelNhwc()
308 auto outputShift = std::vector<int32_t>(outputDepth, .0f); in convQuant8PerChannelNhwc()
310 for (int i = 0; i < outputDepth; ++i) { in convQuant8PerChannelNhwc()
332 for (uint32_t d = 0; d < outputDepth; d++) { in convQuant8PerChannelNhwc()
366 outPtr += outputDepth; in convQuant8PerChannelNhwc()
|