Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLogSoftmax.cpp45 const uint32_t axisSize = getSizeOfDimension(shape, axis); in compute() local
52 T maxValue = input[outer * axisSize * innerSize + inner]; in compute()
53 for (uint32_t i = 1; i < axisSize; ++i) { in compute()
54 maxValue = std::max(maxValue, input[(outer * axisSize + i) * innerSize + inner]); in compute()
58 for (uint32_t i = 0; i < axisSize; ++i) { in compute()
60 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta)); in compute()
64 for (uint32_t i = 0; i < axisSize; ++i) { in compute()
65 output[(outer * axisSize + i) * innerSize + inner] = in compute()
66 (input[(outer * axisSize + i) * innerSize + inner] - maxValue) * beta - in compute()
DL2Normalization.cpp52 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); 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()
79 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8Impl() local
83 const uint8_t* inputBeg = inputData + outer * axisSize * innerSize; in l2normQuant8Impl()
84 const uint8_t* inputEnd = inputBeg + axisSize * innerSize; in l2normQuant8Impl()
85 uint8_t* outputBeg = outputData + outer * axisSize * innerSize; in l2normQuant8Impl()
111 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8SignedImpl() local
115 const int8_t* inputBeg = inputData + outer * axisSize * innerSize; in l2normQuant8SignedImpl()
[all …]
DArgMinMax.cpp36 const int axisSize = getSizeOfDimension(inputShape, axis); in argMinMaxImpl() local
41 auto minMaxValue = inputData[outer * axisSize * innerSize + inner]; in argMinMaxImpl()
43 for (int i = 1; i < axisSize; ++i) { in argMinMaxImpl()
44 const auto& value = inputData[(outer * axisSize + i) * innerSize + inner]; in argMinMaxImpl()
DSoftmax.cpp55 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in softmaxSlowFloat32() local
59 const float* inputBeg = inputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
60 const float* inputEnd = inputBeg + axisSize * innerSize; in softmaxSlowFloat32()
61 float* outputBeg = outputData + outer * axisSize * innerSize; in softmaxSlowFloat32()
130 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in softmaxQuant8Impl() local
134 const T* inputBeg = inputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
135 const T* inputEnd = inputBeg + axisSize * innerSize; in softmaxQuant8Impl()
136 T* outputBeg = outputData + outer * axisSize * innerSize; in softmaxQuant8Impl()
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()
DChannelShuffle.cpp44 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in eval() local
47 const uint32_t groupSize = axisSize / numGroups; in eval()
50 const T* inputBase = inputData + outer * axisSize * innerSize + inner; in eval()
51 T* outputBase = outputData + outer * axisSize * innerSize + inner; in eval()
DLocalResponseNormalization.cpp56 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in localResponseNormFloat32Impl() local
60 const float* inputBase = inputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
61 float* outputBase = outputData + outer * axisSize * innerSize; in localResponseNormFloat32Impl()
63 for (int32_t i = 0; i < axisSize; i++) { in localResponseNormFloat32Impl()
66 const int32_t dEnd = std::min(static_cast<int32_t>(axisSize), i + radius + 1); in localResponseNormFloat32Impl()
DGather.cpp46 const auto axisSize = getSizeOfDimension(inputShape, axis); in eval() local
54 NN_RET_CHECK_LT(inputIndex, axisSize); in eval()
56 inputData + (outer * axisSize + inputIndex) * innerSize, in eval()
/frameworks/minikin/libs/minikin/
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()
/frameworks/ml/nn/common/
DOperationsUtils.cpp601 int32_t axisSize = static_cast<int32_t>(getSizeOfDimension(axisShape, 0)); in meanPrepare() local
608 for (int32_t axisIdx = 0; axisIdx < axisSize; ++axisIdx) { in meanPrepare()
623 int32_t numReduceAxis = axisSize; in meanPrepare()
624 for (int32_t i = 0; i < axisSize; ++i) { in meanPrepare()
646 for (int32_t axisIdx = 0; axisIdx < axisSize; ++axisIdx) { in meanPrepare()