Searched refs:aFlatIndex (Results 1 – 5 of 5) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | MaximumMinimum.cpp | 46 uint32_t aFlatIndex; in evalGeneric() local
47 NN_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in evalGeneric()
51 outputData[outputFlatIndex] = isMinimum ? std::min(aData[aFlatIndex], bData[bFlatIndex]) in evalGeneric()
52 : std::max(aData[aFlatIndex], bData[bFlatIndex]); in evalGeneric()
72 uint32_t aFlatIndex; in evalQuant8() local
73 NN_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in evalQuant8()
77 T aValue = requantize<T>(aData[aFlatIndex], aShape, outputShape); in evalQuant8()
|
| H A D | LogicalAndOr.cpp | 49 uint32_t aFlatIndex; in compute() local
50 NN_RET_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in compute()
54 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in compute()
|
| H A D | PRelu.cpp | 57 uint32_t aFlatIndex; in eval() local
58 NN_RET_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in eval()
62 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in eval()
|
| H A D | Comparisons.cpp | 51 uint32_t aFlatIndex; in compute() local
52 NN_RET_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in compute()
58 const float realA = (aData[aFlatIndex] - aShape.offset) * aShape.scale; in compute()
62 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in compute()
|
| H A D | Broadcast.cpp | 223 uint32_t aFlatIndex; in executeInt32() local
224 NN_RET_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in executeInt32()
228 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in executeInt32()
|