Searched refs:outputShapeIndexed (Results 1 – 6 of 6) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | MaximumMinimum.cpp | 39 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric() local
45 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
54 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
65 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalQuant8() local
71 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalQuant8()
82 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalQuant8()
|
| H A D | Pow.cpp | 38 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric() local
44 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
53 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
|
| H A D | LogicalAndOr.cpp | 43 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute() local
48 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in compute()
56 NN_RET_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in compute()
|
| H A D | PRelu.cpp | 51 IndexedShapeWrapper outputShapeIndexed(outputShape); in eval() local
56 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in eval()
64 NN_RET_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in eval()
|
| H A D | Comparisons.cpp | 45 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute() local
50 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in compute()
65 NN_RET_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in compute()
|
| H A D | Broadcast.cpp | 217 IndexedShapeWrapper outputShapeIndexed(outputShape); in executeInt32() local
222 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in executeInt32()
230 NN_RET_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in executeInt32()
|