Searched refs:axisSize (Results 1 – 9 of 9) sorted by relevance
/packages/modules/NeuralNetworks/common/operations/ |
D | LogSoftmax.cpp | 44 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()
|
D | L2Normalization.cpp | 53 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 …]
|
D | Softmax.cpp | 57 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()
|
D | ArgMinMax.cpp | 32 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()
|
D | LocalResponseNormalization.cpp | 57 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()
|
D | ChannelShuffle.cpp | 41 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()
|
D | SimpleMath.cpp | 55 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()
|
D | Gather.cpp | 43 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/ |
D | OperationsUtils.cpp | 599 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()
|