Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DL2Normalization.cpp44 const uint32_t innerSize = in l2normFloat32Impl() local
47 const float* inputBeg = inputData + outer * axisSize * innerSize; in l2normFloat32Impl()
48 const float* inputEnd = inputBeg + axisSize * innerSize; in l2normFloat32Impl()
49 float* outputBeg = outputData + outer * axisSize * innerSize; in l2normFloat32Impl()
50 for (uint32_t inner = 0; inner < innerSize; ++inner, ++inputBeg, ++inputEnd, ++outputBeg) { in l2normFloat32Impl()
52 for (const float* p = inputBeg; p < inputEnd; p += innerSize) { in l2normFloat32Impl()
58 for (const float* p = inputBeg; p < inputEnd; p += innerSize, pOut += innerSize) { in l2normFloat32Impl()
71 const uint32_t innerSize = in l2normQuant8Impl() local
74 const uint8_t* inputBeg = inputData + outer * axisSize * innerSize; in l2normQuant8Impl()
75 const uint8_t* inputEnd = inputBeg + axisSize * innerSize; in l2normQuant8Impl()
[all …]
DSoftmax.cpp47 const uint32_t innerSize = in softmaxSlowFloat32() local
50 const float* inputBeg = inputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
51 const float* inputEnd = inputBeg + axisSize * innerSize; in softmaxSlowFloat32()
52 float* outputBeg = outputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
53 for (uint32_t inner = 0; inner < innerSize; ++inner, ++inputBeg, ++inputEnd, ++outputBeg) { in softmaxSlowFloat32()
56 for (const float* p = inputBeg; p < inputEnd; p += innerSize) { in softmaxSlowFloat32()
61 for (const float* p = inputBeg; p < inputEnd; p += innerSize) { in softmaxSlowFloat32()
66 for (const float* p = inputBeg; p < inputEnd; p += innerSize, pOut += innerSize) { in softmaxSlowFloat32()
121 const uint32_t innerSize = in softmaxQuant8Impl() local
124 const uint8_t* inputBeg = inputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
[all …]
DNormalization.cpp36 const uint32_t innerSize = in localResponseNormFloat32Impl() local
39 const float* inputBase = inputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
40 float* outputBase = outputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
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()
DArgMinMax.cpp35 const int innerSize = getNumberOfElements( in argMinMaxImpl() local
38 for (int inner = 0; inner < innerSize; ++inner) { in argMinMaxImpl()
39 auto minMaxValue = inputData[outer * axisSize * innerSize + inner]; in argMinMaxImpl()
43 inputData[(outer * axisSize + i) * innerSize + inner]; in argMinMaxImpl()
50 outputData[outer * innerSize + inner] = minMaxIndex; in argMinMaxImpl()
DLogSoftmax.cpp44 const uint32_t innerSize = getNumberOfElements(shape, axis + 1, getNumberOfDimensions(shape)); in compute() local
46 for (uint32_t inner = 0; inner < innerSize; ++inner) { in compute()
50 T maxValue = input[outer * axisSize * innerSize + inner]; in compute()
52 maxValue = std::max(maxValue, input[(outer * axisSize + i) * innerSize + inner]); in compute()
58 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta)); in compute()
63 output[(outer * axisSize + i) * innerSize + inner] = in compute()
64 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta - in compute()
DChannelShuffle.cpp43 const uint32_t innerSize = in eval() local
47 for (uint32_t inner = 0; inner < innerSize; ++inner) { in eval()
48 const T* inputBase = inputData + outer * axisSize * innerSize + inner; in eval()
49 T* outputBase = outputData + outer * axisSize * innerSize + inner; in eval()
52 j++, outputBase += innerSize) { in eval()
53 *outputBase = inputBase[innerSize * (i + j * groupSize)]; in eval()
DGather.cpp45 const auto innerSize = in eval() local
53 std::memcpy(outputData + (outer * indicesCount + outputIndex) * innerSize, in eval()
54 inputData + (outer * axisSize + inputIndex) * innerSize, in eval()
55 sizeof(T) * innerSize); in eval()