Searched refs:outputFlatIndex (Results 1 – 6 of 6) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | MaximumMinimum.cpp | 44 uint32_t outputFlatIndex; in evalGeneric() local
45 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
51 outputData[outputFlatIndex] = isMinimum ? std::min(aData[aFlatIndex], bData[bFlatIndex]) in evalGeneric()
70 uint32_t outputFlatIndex; in evalQuant8() local
71 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalQuant8()
79 outputData[outputFlatIndex] = in evalQuant8()
|
| D | Pow.cpp | 43 uint32_t outputFlatIndex; in evalGeneric() local
44 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
50 outputData[outputFlatIndex] = std::pow(static_cast<float>(baseData[baseFlatIndex]), in evalGeneric()
|
| D | LogicalAndOr.cpp | 41 uint32_t outputFlatIndex; in compute() local
42 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in compute()
48 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in compute()
|
| D | Comparisons.cpp | 43 uint32_t outputFlatIndex; in compute() local
44 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in compute()
54 outputData[outputFlatIndex] = func(realA, realB); in compute()
56 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in compute()
|
| D | PRelu.cpp | 54 uint32_t outputFlatIndex; in eval() local
55 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in eval()
61 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in eval()
|
| D | Broadcast.cpp | 221 uint32_t outputFlatIndex; in executeInt32() local
222 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in executeInt32()
228 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in executeInt32()
|