chore: import upstream snapshot with attribution

2026-07-13 13:33:03 +08:00
commit 5b57521aa1
8226 changed files with 3425766 additions and 0 deletions
@@ -0,0 +1,86 @@
+//
+//  NNAPIBinary.cpp
+//  MNN
+//
+//  Created by MNN on 2022/09/05.
+//  Copyright © 2018, Alibaba Group Holding Limited
+//
+
+#include "NNAPIBinary.hpp"
+
+namespace MNN {
+
+
+NNAPIBinary::NNAPIBinary(MNN::Backend *b, const MNN::Op *op, const std::vector<Tensor *> &inputs, const std::vector<MNN::Tensor *> &outputs) : NNAPICommonExecution(b, op) {
+}
+
+ErrorCode NNAPIBinary::onResize(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs) {
+    MNN_ASSERT(inputs.size() == 2 && outputs.size() == 1);
+    std::map<BinaryOpOperation, int> binary_map {
+        {BinaryOpOperation_ADD, ANEURALNETWORKS_ADD},
+        {BinaryOpOperation_SUB, ANEURALNETWORKS_SUB},
+        {BinaryOpOperation_MUL, ANEURALNETWORKS_MUL},
+        {BinaryOpOperation_DIV, ANEURALNETWORKS_DIV},
+        {BinaryOpOperation_POW, ANEURALNETWORKS_POW},
+        {BinaryOpOperation_REALDIV, ANEURALNETWORKS_DIV},
+        {BinaryOpOperation_MINIMUM, ANEURALNETWORKS_MINIMUM},
+        {BinaryOpOperation_MAXIMUM, ANEURALNETWORKS_MAXIMUM},
+        {BinaryOpOperation_GREATER, ANEURALNETWORKS_GREATER},
+        {BinaryOpOperation_GREATER_EQUAL, ANEURALNETWORKS_GREATER_EQUAL},
+        {BinaryOpOperation_LESS, ANEURALNETWORKS_LESS},
+        {BinaryOpOperation_FLOORDIV, -1},
+        {BinaryOpOperation_SquaredDifference, -1},
+        {BinaryOpOperation_LESS_EQUAL, ANEURALNETWORKS_LESS_EQUAL},
+        {BinaryOpOperation_FLOORMOD, -1},
+        {BinaryOpOperation_EQUAL, ANEURALNETWORKS_EQUAL},
+        {BinaryOpOperation_MOD, -1},
+        {BinaryOpOperation_ATAN2, -1},
+        {BinaryOpOperation_LOGICALOR, ANEURALNETWORKS_LOGICAL_OR},
+        {BinaryOpOperation_NOTEQUAL, ANEURALNETWORKS_NOT_EQUAL},
+        {BinaryOpOperation_BITWISE_AND, -1},
+        {BinaryOpOperation_BITWISE_OR, -1},
+        {BinaryOpOperation_BITWISE_XOR, -1},
+        {BinaryOpOperation_LOGICALXOR, -1},
+        {BinaryOpOperation_LEFTSHIFT, -1},
+        {BinaryOpOperation_RIGHTSHIFT, -1}
+    };
+    BinaryOpOperation binaryType;
+    auto fuseType = ANEURALNETWORKS_FUSED_NONE;
+    if (mOp->type() == OpType_BinaryOp) {
+        binaryType = static_cast<BinaryOpOperation>(mOp->main_as_BinaryOp()->opType());
+        if (mOp->main_as_BinaryOp()->activationType() == 1) {
+            fuseType = ANEURALNETWORKS_FUSED_RELU;
+        }
+    } else if (mOp->type() == OpType_Eltwise) {
+        auto elemType = mOp->main_as_Eltwise()->type();
+        switch (elemType) {
+            case EltwiseType_PROD:
+                binaryType = BinaryOpOperation_MUL;
+                break;
+            case EltwiseType_SUM:
+                binaryType = BinaryOpOperation_ADD;
+                break;
+            case EltwiseType_SUB:
+                binaryType = BinaryOpOperation_SUB;
+                break;
+            case EltwiseType_MAXIMUM:
+                binaryType = BinaryOpOperation_MAXIMUM;
+                break;
+        }
+    }
+    auto iter = binary_map.find(binaryType);
+    if (iter == binary_map.end() || iter->second < 0) {
+        MNN_ERROR("[NNAPI] Binary not support %s\n", MNN::EnumNameBinaryOpOperation(binaryType));
+        return NOT_SUPPORT;
+    }
+    if (TensorUtils::getDescribe(outputs[0])->quantAttr.get()) {
+        outputs[0]->buffer().type = halide_type_of<int8_t>();
+    }
+    auto inputIdxs = getTensorIdxs(inputs);
+    inputIdxs.push_back(buildScalar(fuseType));
+    return buildOperation(iter->second, inputIdxs, getTensorIdxs(outputs));
+}
+
+REGISTER_NNAPI_OP_CREATOR(NNAPIBinary, OpType_BinaryOp)
+REGISTER_NNAPI_OP_CREATOR(NNAPIBinary, OpType_Eltwise)
+} // namespace MNN