Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DGroupedConv2D.cpp58 const float* inputBase = inputData; in groupedConvFloat32() local
83 sum += filterBase[filterIndex] * inputBase[inputIndex]; in groupedConvFloat32()
97 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvFloat32()
130 const T* inputBase = inputData; in groupedConvQuant8() local
157 (static_cast<int32_t>(inputBase[inputIndex]) + in groupedConvQuant8()
175 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvQuant8()
234 const T* inputBase = inputData; in groupedConvQuant8PerChannel() local
260 (static_cast<int32_t>(inputBase[inputIndex]) + in groupedConvQuant8PerChannel()
279 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvQuant8PerChannel()
DLocalResponseNormalization.cpp60 const float* inputBase = inputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl() local
62 for (uint32_t inner = 0; inner < innerSize; ++inner, ++inputBase, ++outputBase) { in localResponseNormFloat32Impl()
69 float val = inputBase[d * innerSize]; in localResponseNormFloat32Impl()
73 outputBase[i * innerSize] = inputBase[i * innerSize] * multiplier; in localResponseNormFloat32Impl()
DChannelShuffle.cpp50 const T* inputBase = inputData + outer * axisSize * innerSize + inner; in eval() local
55 *outputBase = inputBase[innerSize * (i + j * groupSize)]; in eval()
DTransposeConv2D.cpp137 const float* inputBase = inputData; in transposeConvNhwc() local
156 outputBase[outputIndex] += inputBase[d] * filterBase[d]; in transposeConvNhwc()
163 inputBase += inputDepth; in transposeConvNhwc()
DDepthwiseConv2D.cpp321 const T* inputBase = inputData; in depthwiseConvQuant8PerChannelNhwc() local
347 (static_cast<int32_t>(inputBase[inputIndex]) + in depthwiseConvQuant8PerChannelNhwc()
364 inputBase += inputHeight * inputWidth * inputDepth; in depthwiseConvQuant8PerChannelNhwc()
DConv2D.cpp394 const uint8_t* inputBase = inputData; in convQuant8PerChannelNhwc() local
421 (static_cast<int32_t>(inputBase[inputIndex]) + in convQuant8PerChannelNhwc()
438 inputBase += inputHeight * inputWidth * inputDepth; in convQuant8PerChannelNhwc()