Searched refs:outputShapeIndexed (Results 1 – 6 of 6) sorted by relevance
/packages/modules/NeuralNetworks/common/cpu_operations/ |
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()
|
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()
|
D | LogicalAndOr.cpp | 37 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute() local 42 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in compute() 50 NN_RET_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in compute()
|
D | PRelu.cpp | 50 IndexedShapeWrapper outputShapeIndexed(outputShape); in eval() local 55 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in eval() 63 NN_RET_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in eval()
|
D | Comparisons.cpp | 39 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute() local 44 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in compute() 59 NN_RET_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in compute()
|
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()
|