Home
last modified time | relevance | path

Searched refs:innerSize (Results 1 – 7 of 7) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DL2Normalization.cpp52 const uint32_t innerSize = in l2normFloat32Impl() local
55 const float* inputBeg = inputData + outer * axisSize * innerSize; in l2normFloat32Impl()
56 const float* inputEnd = inputBeg + axisSize * innerSize; in l2normFloat32Impl()
57 float* outputBeg = outputData + outer * axisSize * innerSize; in l2normFloat32Impl()
58 for (uint32_t inner = 0; inner < innerSize; ++inner, ++inputBeg, ++inputEnd, ++outputBeg) { in l2normFloat32Impl()
60 for (const float* p = inputBeg; p < inputEnd; p += innerSize) { in l2normFloat32Impl()
66 for (const float* p = inputBeg; p < inputEnd; p += innerSize, pOut += innerSize) { in l2normFloat32Impl()
79 const uint32_t innerSize = in l2normQuant8Impl() local
82 const uint8_t* inputBeg = inputData + outer * axisSize * innerSize; in l2normQuant8Impl()
83 const uint8_t* inputEnd = inputBeg + axisSize * innerSize; in l2normQuant8Impl()
[all …]
DSoftmax.cpp55 const uint32_t innerSize = in softmaxSlowFloat32() local
58 const float* inputBeg = inputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
59 const float* inputEnd = inputBeg + axisSize * innerSize; in softmaxSlowFloat32()
60 float* outputBeg = outputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
61 for (uint32_t inner = 0; inner < innerSize; ++inner, ++inputBeg, ++inputEnd, ++outputBeg) { in softmaxSlowFloat32()
64 for (const float* p = inputBeg; p < inputEnd; p += innerSize) { in softmaxSlowFloat32()
69 for (const float* p = inputBeg; p < inputEnd; p += innerSize) { in softmaxSlowFloat32()
74 for (const float* p = inputBeg; p < inputEnd; p += innerSize, pOut += innerSize) { in softmaxSlowFloat32()
130 const uint32_t innerSize = in softmaxQuant8Impl() local
133 const T* inputBeg = inputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
[all …]
DLogSoftmax.cpp37 const uint32_t innerSize = getNumberOfElements(shape, axis + 1, getNumberOfDimensions(shape)); in compute() local
39 for (uint32_t inner = 0; inner < innerSize; ++inner) { in compute()
43 T maxValue = input[outer * axisSize * innerSize + inner]; in compute()
45 maxValue = std::max(maxValue, input[(outer * axisSize + i) * innerSize + inner]); in compute()
51 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta)); in compute()
56 output[(outer * axisSize + i) * innerSize + inner] = in compute()
57 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta - in compute()
DChannelShuffle.cpp34 const uint32_t innerSize = in eval() local
38 for (uint32_t inner = 0; inner < innerSize; ++inner) { in eval()
39 const T* inputBase = inputData + outer * axisSize * innerSize + inner; in eval()
40 T* outputBase = outputData + outer * axisSize * innerSize + inner; in eval()
43 j++, outputBase += innerSize) { in eval()
44 *outputBase = inputBase[innerSize * (i + j * groupSize)]; in eval()
DArgMinMax.cpp35 const int innerSize = in argMinMaxImpl() local
38 for (int inner = 0; inner < innerSize; ++inner) { in argMinMaxImpl()
39 auto minMaxValue = inputData[outer * axisSize * innerSize + inner]; in argMinMaxImpl()
42 const auto& value = inputData[(outer * axisSize + i) * innerSize + inner]; in argMinMaxImpl()
48 outputData[outer * innerSize + inner] = minMaxIndex; in argMinMaxImpl()
DLocalResponseNormalization.cpp52 const uint32_t innerSize = in localResponseNormFloat32Impl() local
55 const float* inputBase = inputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
56 float* outputBase = outputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
57 for (uint32_t inner = 0; inner < innerSize; ++inner, ++inputBase, ++outputBase) { in localResponseNormFloat32Impl()
64 float val = inputBase[d * innerSize]; in localResponseNormFloat32Impl()
68 outputBase[i * innerSize] = inputBase[i * innerSize] * multiplier; in localResponseNormFloat32Impl()
DGather.cpp35 const auto innerSize = in eval() local
43 std::memcpy(outputData + (outer * indicesCount + outputIndex) * innerSize, in eval()
44 inputData + (outer * axisSize + inputIndex) * innerSize, in eval()
45 sizeof(T) * innerSize); in eval()