Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DL2Normalization.cpp53 const uint32_t innerSize = in l2normFloat32Impl() local
56 const float* inputBeg = inputData + outer * axisSize * innerSize; in l2normFloat32Impl()
57 const float* inputEnd = inputBeg + axisSize * innerSize; in l2normFloat32Impl()
58 float* outputBeg = outputData + outer * axisSize * innerSize; in l2normFloat32Impl()
59 for (uint32_t inner = 0; inner < innerSize; ++inner, ++inputBeg, ++inputEnd, ++outputBeg) { in l2normFloat32Impl()
61 for (const float* p = inputBeg; p < inputEnd; p += innerSize) { in l2normFloat32Impl()
67 for (const float* p = inputBeg; p < inputEnd; p += innerSize, pOut += innerSize) { in l2normFloat32Impl()
80 const uint32_t innerSize = in l2normQuant8Impl() local
83 const uint8_t* inputBeg = inputData + outer * axisSize * innerSize; in l2normQuant8Impl()
84 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 …]
DArgMinMax.cpp37 const int innerSize = in argMinMaxImpl() local
40 for (int inner = 0; inner < innerSize; ++inner) { in argMinMaxImpl()
41 auto minMaxValue = inputData[outer * axisSize * innerSize + inner]; in argMinMaxImpl()
44 const auto& value = inputData[(outer * axisSize + i) * innerSize + inner]; in argMinMaxImpl()
50 outputData[outer * innerSize + inner] = minMaxIndex; in argMinMaxImpl()
DLogSoftmax.cpp46 const uint32_t innerSize = getNumberOfElements(shape, axis + 1, getNumberOfDimensions(shape)); in compute() local
48 for (uint32_t inner = 0; inner < innerSize; ++inner) { in compute()
52 T maxValue = input[outer * axisSize * innerSize + inner]; in compute()
54 maxValue = std::max(maxValue, input[(outer * axisSize + i) * innerSize + inner]); in compute()
60 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta)); in compute()
65 output[(outer * axisSize + i) * innerSize + inner] = in compute()
66 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta - in compute()
DChannelShuffle.cpp45 const uint32_t innerSize = in eval() local
49 for (uint32_t inner = 0; inner < innerSize; ++inner) { in eval()
50 const T* inputBase = inputData + outer * axisSize * innerSize + inner; in eval()
51 T* outputBase = outputData + outer * axisSize * innerSize + inner; in eval()
54 j++, outputBase += innerSize) { in eval()
55 *outputBase = inputBase[innerSize * (i + j * groupSize)]; in eval()
DLocalResponseNormalization.cpp57 const uint32_t innerSize = in localResponseNormFloat32Impl() local
60 const float* inputBase = inputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
61 float* outputBase = outputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
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()
DGather.cpp47 const auto innerSize = in eval() local
55 std::memcpy(outputData + (outer * indicesCount + outputIndex) * innerSize, in eval()
56 inputData + (outer * axisSize + inputIndex) * innerSize, in eval()
57 sizeof(T) * innerSize); in eval()