Searched refs:outputShapeIndexed (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | MaximumMinimum.cpp | 35 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric() local
41 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
50 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
61 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalQuant8() local
67 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalQuant8()
78 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalQuant8()
|
| D | Pow.cpp | 36 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric() local
42 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
51 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
|
| D | LogicalAndOr.cpp | 41 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute() local
46 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in compute()
54 NN_RET_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in compute()
|
| D | PRelu.cpp | 45 IndexedShapeWrapper outputShapeIndexed(outputShape); in eval() local
50 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in eval()
58 NN_RET_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in eval()
|
| D | Comparisons.cpp | 43 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute() local
48 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in compute()
62 NN_RET_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in compute()
|