Home
last modified time | relevance | path

Searched refs:outputShapeIndexed (Results 1 – 6 of 6) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DMaximumMinimum.cpp39 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()
DPow.cpp38 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric() local
44 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
53 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
DLogicalAndOr.cpp37 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()
DPRelu.cpp50 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()
DComparisons.cpp39 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()
DBroadcast.cpp217 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()