Home
last modified time | relevance | path

Searched refs:axisSize (Results 1 – 9 of 9) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DLogSoftmax.cpp36 const uint32_t axisSize = getSizeOfDimension(shape, axis); in compute() local
43 T maxValue = input[outer * axisSize * innerSize + inner]; in compute()
44 for (uint32_t i = 1; i < axisSize; ++i) { in compute()
45 maxValue = std::max(maxValue, input[(outer * axisSize + i) * innerSize + inner]); in compute()
49 for (uint32_t i = 0; i < axisSize; ++i) { in compute()
51 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta)); in compute()
55 for (uint32_t i = 0; i < axisSize; ++i) { in compute()
56 output[(outer * axisSize + i) * innerSize + inner] = in compute()
57 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta - in compute()
DL2Normalization.cpp51 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); 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()
78 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8Impl() local
82 const uint8_t* inputBeg = inputData + outer * axisSize * innerSize; in l2normQuant8Impl()
83 const uint8_t* inputEnd = inputBeg + axisSize * innerSize; in l2normQuant8Impl()
84 uint8_t* outputBeg = outputData + outer * axisSize * innerSize; in l2normQuant8Impl()
110 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8SignedImpl() local
114 const int8_t* inputBeg = inputData + outer * axisSize * innerSize; in l2normQuant8SignedImpl()
[all …]
DArgMinMax.cpp34 const int axisSize = getSizeOfDimension(inputShape, axis); in argMinMaxImpl() local
39 auto minMaxValue = inputData[outer * axisSize * innerSize + inner]; in argMinMaxImpl()
41 for (int i = 1; i < axisSize; ++i) { in argMinMaxImpl()
42 const auto& value = inputData[(outer * axisSize + i) * innerSize + inner]; in argMinMaxImpl()
DSoftmax.cpp54 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); 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()
129 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in softmaxQuant8Impl() local
133 const T* inputBeg = inputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
134 const T* inputEnd = inputBeg + axisSize * innerSize; in softmaxQuant8Impl()
135 T* outputBeg = outputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
DChannelShuffle.cpp33 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in eval() local
36 const uint32_t groupSize = axisSize / numGroups; in eval()
39 const T* inputBase = inputData + outer * axisSize * innerSize + inner; in eval()
40 T* outputBase = outputData + outer * axisSize * innerSize + inner; in eval()
DLocalResponseNormalization.cpp51 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in localResponseNormFloat32Impl() local
55 const float* inputBase = inputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
56 float* outputBase = outputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
58 for (int32_t i = 0; i < static_cast<int32_t>(axisSize); i++) { in localResponseNormFloat32Impl()
61 const int32_t dEnd = std::min(static_cast<int32_t>(axisSize), i + radius + 1); in localResponseNormFloat32Impl()
DSimpleMath.cpp61 int32_t axisSize = static_cast<int32_t>(getSizeOfDimension(axisShape, 0)); in meanGeneric() local
62 int32_t* resolvedAxis = new int32_t[axisSize]; in meanGeneric()
75 getNumberOfDimensions(outputShape), axis, axisSize, keepDims, scratchBuffer, in meanGeneric()
DGather.cpp34 const auto axisSize = getSizeOfDimension(inputShape, axis); in eval() local
42 NN_RET_CHECK_LT(inputIndex, axisSize); in eval()
44 inputData + (outer * axisSize + inputIndex) * innerSize, in eval()
/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp508 int32_t axisSize = static_cast<int32_t>(getSizeOfDimension(axisShape, 0)); in meanPrepare() local
515 for (int32_t axisIdx = 0; axisIdx < axisSize; ++axisIdx) { in meanPrepare()
530 int32_t numReduceAxis = axisSize; in meanPrepare()
531 for (int32_t i = 0; i < axisSize; ++i) { in meanPrepare()
553 for (int32_t axisIdx = 0; axisIdx < axisSize; ++axisIdx) { in meanPrepare()