Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A 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()
H A 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
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
117 const int8_t* inputEnd = inputBeg + axisSize * innerSize; in l2normQuant8SignedImpl()
118 int8_t* outputBeg = outputData + outer * axisSize * innerSize; in l2normQuant8SignedImpl()
[all …]
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
/aosp12/frameworks/minikin/libs/minikin/
H A DFontUtils.cpp64 const uint32_t axisSize = readU16(fvar_data, kAxisSizeOffset); in analyzeAxes() local
66 if (majorVersion != 1 || minorVersion != 0 || axisOffset != 0x10 || axisSize != 0x14) { in analyzeAxes()
69 if (fvar_size < axisOffset + axisSize * axisCount) { in analyzeAxes()
70 if (axisOffset > axisSize) { in analyzeAxes()
76 size_t axisRecordOffset = axisOffset + i * axisSize; in analyzeAxes()
/aosp12/packages/modules/NeuralNetworks/common/
H A 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()