Searched refs:broadcastedIndexToFlatIndex (Results 1 – 8 of 8) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | MaximumMinimum.cpp | 47 NN_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in evalGeneric()
49 NN_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in evalGeneric()
73 NN_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in evalQuant8()
75 NN_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in evalQuant8()
|
| D | Pow.cpp | 46 NN_CHECK(baseShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &baseFlatIndex)); in evalGeneric()
48 NN_CHECK(exponentShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &exponentFlatIndex)); in evalGeneric()
|
| D | LogicalAndOr.cpp | 44 NN_RET_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in compute()
46 NN_RET_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in compute()
|
| D | PRelu.cpp | 57 NN_RET_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in eval()
59 NN_RET_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in eval()
|
| D | Comparisons.cpp | 46 NN_RET_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in compute()
48 NN_RET_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in compute()
|
| D | Broadcast.cpp | 224 NN_RET_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in executeInt32()
226 NN_RET_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in executeInt32()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | IndexedShapeWrapper.h | 55 bool broadcastedIndexToFlatIndex(const std::vector<uint32_t>& index, uint32_t* flatIndex) const;
|
| /packages/modules/NeuralNetworks/common/ |
| D | IndexedShapeWrapper.cpp | 72 bool IndexedShapeWrapper::broadcastedIndexToFlatIndex(const std::vector<uint32_t>& index, in broadcastedIndexToFlatIndex() function in android::nn::IndexedShapeWrapper
|