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,37 @@
+//
+//  NNAPIScale.cpp
+//  MNN
+//
+//  Created by MNN on 2022/10/25.
+//  Copyright © 2018, Alibaba Group Holding Limited
+//
+
+#include "NNAPIScale.hpp"
+
+namespace MNN {
+
+
+NNAPIScale::NNAPIScale(MNN::Backend *b, const MNN::Op *op, const std::vector<Tensor *> &inputs, const std::vector<MNN::Tensor *> &outputs) : NNAPICommonExecution(b, op) {
+}
+
+ErrorCode NNAPIScale::onResize(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs) {
+    auto scaleParam = mOp->main_as_Scale();
+    auto weight = scaleParam->scaleData();
+    auto bias = scaleParam->biasData();
+    uint32_t channel = scaleParam->channels();    
+    auto inputShape = inputs[0]->shape(); // NCHW
+    int inputSize = inputs[0]->elementSize();
+    std::vector<uint32_t> dims {1, 1, 1, channel};
+    // middle = input * weight
+    auto middleIdx = buildTensor(ANEURALNETWORKS_TENSOR_FLOAT32, inputShape);
+    auto inputIdxs = getTensorIdxs(inputs);
+    inputIdxs.push_back(buildConstant(weight->data(), channel * sizeof(float), ANEURALNETWORKS_TENSOR_FLOAT32, dims));
+    inputIdxs.push_back(buildScalar(ANEURALNETWORKS_FUSED_NONE));
+    buildOperation(ANEURALNETWORKS_MUL, inputIdxs, {middleIdx});
+    // output = middle + bias
+    auto biasIdx = buildConstant(bias->data(), channel * sizeof(float), ANEURALNETWORKS_TENSOR_FLOAT32, dims);
+    return buildOperation(ANEURALNETWORKS_ADD, {middleIdx, biasIdx, buildScalar(ANEURALNETWORKS_FUSED_NONE)}, getTensorIdxs(outputs));
+}
+
+REGISTER_NNAPI_OP_CREATOR(NNAPIScale, OpType_Scale)
+} // namespace MNN