Searched refs:bFlatIndex (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | MaximumMinimum.cpp | 50 uint32_t bFlatIndex; in evalGeneric() local
51 NN_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in evalGeneric()
53 outputData[outputFlatIndex] = isMinimum ? std::min(aData[aFlatIndex], bData[bFlatIndex]) in evalGeneric()
54 : std::max(aData[aFlatIndex], bData[bFlatIndex]); in evalGeneric()
76 uint32_t bFlatIndex; in evalQuant8() local
77 NN_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in evalQuant8()
80 T bValue = requantize<T>(bData[bFlatIndex], bShape, outputShape); in evalQuant8()
|
| D | LogicalAndOr.cpp | 51 uint32_t bFlatIndex; in compute() local
52 NN_RET_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in compute()
54 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in compute()
|
| D | PRelu.cpp | 59 uint32_t bFlatIndex; in eval() local
60 NN_RET_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in eval()
62 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in eval()
|
| D | Comparisons.cpp | 56 uint32_t bFlatIndex; in compute() local
57 NN_RET_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in compute()
62 const float realB = (bData[bFlatIndex] - bShape.offset) * bShape.scale; in compute()
65 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in compute()
|
| D | Broadcast.cpp | 222 uint32_t bFlatIndex; in executeInt32() local
223 NN_RET_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in executeInt32()
225 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in executeInt32()
|