Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/operations/
DLogSoftmax.cpp44 const uint32_t axisSize = getSizeOfDimension(shape, axis); in compute() local
51 T maxValue = input[outer * axisSize * innerSize + inner]; in compute()
52 for (uint32_t i = 1; i < axisSize; ++i) { in compute()
53 maxValue = std::max(maxValue, input[(outer * axisSize + i) * innerSize + inner]); in compute()
57 for (uint32_t i = 0; i < axisSize; ++i) { in compute()
59 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta)); in compute()
63 for (uint32_t i = 0; i < axisSize; ++i) { in compute()
64 output[(outer * axisSize + i) * innerSize + inner] = in compute()
65 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta - in compute()
DL2Normalization.cpp53 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normFloat32Impl() local
57 const float* inputBeg = inputData + outer * axisSize * innerSize; in l2normFloat32Impl()
58 const float* inputEnd = inputBeg + axisSize * innerSize; in l2normFloat32Impl()
59 float* outputBeg = outputData + outer * axisSize * innerSize; in l2normFloat32Impl()
80 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8Impl() local
84 const uint8_t* inputBeg = inputData + outer * axisSize * innerSize; in l2normQuant8Impl()
85 const uint8_t* inputEnd = inputBeg + axisSize * innerSize; in l2normQuant8Impl()
86 uint8_t* outputBeg = outputData + outer * axisSize * innerSize; in l2normQuant8Impl()
112 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8SignedImpl() local
116 const int8_t* inputBeg = inputData + outer * axisSize * innerSize; in l2normQuant8SignedImpl()
[all …]
DSoftmax.cpp57 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in softmaxSlowFloat32() local
61 const float* inputBeg = inputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
62 const float* inputEnd = inputBeg + axisSize * innerSize; in softmaxSlowFloat32()
63 float* outputBeg = outputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
132 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in softmaxQuant8Impl() local
136 const T* inputBeg = inputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
137 const T* inputEnd = inputBeg + axisSize * innerSize; in softmaxQuant8Impl()
138 T* outputBeg = outputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
DArgMinMax.cpp32 const int axisSize = getSizeOfDimension(inputShape, axis); in argMinMaxImpl() local
37 auto minMaxValue = inputData[outer * axisSize * innerSize + inner]; in argMinMaxImpl()
39 for (int i = 1; i < axisSize; ++i) { in argMinMaxImpl()
40 const auto& value = inputData[(outer * axisSize + i) * innerSize + inner]; in argMinMaxImpl()
DLocalResponseNormalization.cpp57 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in localResponseNormFloat32Impl() local
61 const float* inputBase = inputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
62 float* outputBase = outputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
64 for (int32_t i = 0; i < axisSize; i++) { in localResponseNormFloat32Impl()
67 const int32_t dEnd = std::min(static_cast<int32_t>(axisSize), i + radius + 1); in localResponseNormFloat32Impl()
DChannelShuffle.cpp41 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in eval() local
44 const uint32_t groupSize = axisSize / numGroups; in eval()
47 const T* inputBase = inputData + outer * axisSize * innerSize + inner; in eval()
48 T* outputBase = outputData + outer * axisSize * innerSize + inner; in eval()
DSimpleMath.cpp55 int32_t axisSize = static_cast<int32_t>(getSizeOfDimension(axisShape, 0)); in meanGeneric() local
56 int32_t* resolvedAxis = new int32_t[axisSize]; in meanGeneric()
69 getNumberOfDimensions(outputShape), axis, axisSize, keepDims, scratchBuffer, in meanGeneric()
DGather.cpp43 const auto axisSize = getSizeOfDimension(inputShape, axis); in eval() local
51 NN_RET_CHECK_LT(inputIndex, axisSize); in eval()
53 inputData + (outer * axisSize + inputIndex) * innerSize, in eval()
/packages/modules/NeuralNetworks/common/
DOperationsUtils.cpp599 int32_t axisSize = static_cast<int32_t>(getSizeOfDimension(axisShape, 0)); in meanPrepare() local
606 for (int32_t axisIdx = 0; axisIdx < axisSize; ++axisIdx) { in meanPrepare()
621 int32_t numReduceAxis = axisSize; in meanPrepare()
622 for (int32_t i = 0; i < axisSize; ++i) { in meanPrepare()
644 for (int32_t axisIdx = 0; axisIdx < axisSize; ++axisIdx) { in meanPrepare()