Searched refs:indexToFlatIndex (Results 1 – 9 of 9) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | MaximumMinimum.cpp | 45 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
71 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalQuant8()
|
| H A D | Slice.cpp | 74 NN_RET_CHECK(indexedOutput.indexToFlatIndex(outputIndex, &outputOffset)); in evalGeneric()
75 NN_RET_CHECK(indexedInput.indexToFlatIndex(inputIndex, &inputOffset)); in evalGeneric()
|
| H A D | Pow.cpp | 44 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
|
| H A D | LogicalAndOr.cpp | 48 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in compute()
|
| H A D | PRelu.cpp | 56 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in eval()
|
| H A D | Comparisons.cpp | 50 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in compute()
|
| H A D | Broadcast.cpp | 222 NN_RET_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in executeInt32()
|
| /aosp12/packages/modules/NeuralNetworks/common/include/ |
| H A D | IndexedShapeWrapper.h | 43 bool indexToFlatIndex(const std::vector<uint32_t>& index, uint32_t* flatIndex) const;
|
| /aosp12/packages/modules/NeuralNetworks/common/ |
| H A D | IndexedShapeWrapper.cpp | 61 bool IndexedShapeWrapper::indexToFlatIndex(const std::vector<uint32_t>& index, in indexToFlatIndex() function in android::nn::IndexedShapeWrapper
|