Home
last modified time | relevance | path

Searched refs:outputBase (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DNormalization.cpp40 float* outputBase = outputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl() local
41 for (uint32_t inner = 0; inner < innerSize; ++inner, ++inputBase, ++outputBase) { in localResponseNormFloat32Impl()
52 outputBase[i * innerSize] = inputBase[i * innerSize] * multiplier; in localResponseNormFloat32Impl()
DChannelShuffle.cpp49 T* outputBase = outputData + outer * axisSize * innerSize + inner; in eval() local
52 j++, outputBase += innerSize) { in eval()
53 *outputBase = inputBase[innerSize * (i + j * groupSize)]; in eval()
DTransposeConv2D.cpp128 float* outputBase = outputData; in transposeConvNhwc() local
146 outputBase[outputIndex] += inputBase[d] * filterBase[d]; in transposeConvNhwc()
156 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvNhwc()
214 int32_t* outputBase = tempBuffer; in transposeConvNhwc() local
234 outputBase[outputIndex] += in transposeConvNhwc()
247 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvNhwc()
355 int32_t* outputBase = tempBuffer; in transposeConvQuant8PerChannelNhwc() local
375 outputBase[outputIndex] += in transposeConvQuant8PerChannelNhwc()
387 outputBase += outputHeight * outputWidth * outputDepth; in transposeConvQuant8PerChannelNhwc()