Home
last modified time | relevance | path

Searched refs:inputBase (Results 1 – 6 of 6) sorted by relevance

/frameworks/ml/nn/common/operations/
DNormalization.cpp39 const float* inputBase = inputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl() local
41 for (uint32_t inner = 0; inner < innerSize; ++inner, ++inputBase, ++outputBase) { in localResponseNormFloat32Impl()
48 float val = inputBase[d * innerSize]; in localResponseNormFloat32Impl()
52 outputBase[i * innerSize] = inputBase[i * innerSize] * multiplier; in localResponseNormFloat32Impl()
DGroupedConv2D.cpp54 const float* inputBase = inputData; in groupedConvFloat32() local
79 sum += filterBase[filterIndex] * inputBase[inputIndex]; in groupedConvFloat32()
93 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvFloat32()
125 const uint8_t* inputBase = inputData; in groupedConvQuant8() local
152 (static_cast<int32_t>(inputBase[inputIndex]) + in groupedConvQuant8()
170 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvQuant8()
211 const uint8_t* inputBase = inputData; in groupedConvQuant8PerChannel() local
237 (static_cast<int32_t>(inputBase[inputIndex]) + in groupedConvQuant8PerChannel()
256 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvQuant8PerChannel()
DChannelShuffle.cpp48 const T* inputBase = inputData + outer * axisSize * innerSize + inner; in eval() local
53 *outputBase = inputBase[innerSize * (i + j * groupSize)]; in eval()
DDepthwiseConv2D.cpp195 const uint8_t* inputBase = inputData; in depthwiseConvQuant8PerChannel() local
221 (static_cast<int32_t>(inputBase[inputIndex]) + in depthwiseConvQuant8PerChannel()
238 inputBase += inputHeight * inputWidth * inputDepth; in depthwiseConvQuant8PerChannel()
DTransposeConv2D.cpp127 const float* inputBase = inputData; in transposeConvNhwc() local
146 outputBase[outputIndex] += inputBase[d] * filterBase[d]; in transposeConvNhwc()
153 inputBase += inputDepth; in transposeConvNhwc()
DConv2D.cpp325 const uint8_t* inputBase = inputData; in convQuant8PerChannelNhwc() local
352 (static_cast<int32_t>(inputBase[inputIndex]) + in convQuant8PerChannelNhwc()
369 inputBase += inputHeight * inputWidth * inputDepth; in convQuant8PerChannelNhwc()