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,53 @@
+//
+//  ReshapeGrad.cpp
+//  MNN
+//
+//  Created by MNN on 2019/04/22.
+//  Copyright © 2018, Alibaba Group Holding Limited
+//
+
+#include "ReshapeGrad.hpp"
+#include "core/Macro.h"
+using namespace std;
+namespace MNN {
+using namespace MNN::Express;
+
+class ReshapeGrad : public OpGrad {
+public:
+    virtual std::vector<Express::VARP> onGrad(Express::EXPRP expr,
+                                              const std::vector<Express::VARP>& backwardOutput) override {
+        auto inputs = expr->inputs();
+        std::vector<VARP> result(inputs.size(), nullptr);
+        auto info = inputs[0]->getInfo();
+        if (nullptr == info) {
+            return {};
+        }
+        if (info->order != NC4HW4) {
+            auto shape = _Shape(inputs[0]);
+            // Create Reshape Op
+            result[0] = _Reshape(backwardOutput[0], shape);
+        } else {
+            // NC4HW4 don't support dynamic shape grad
+            // Create Reshape Op
+            // result[0] = _Reshape(backwardOutput[0], _Const(info->dim.data(), {(int)info->dim.size()}, NCHW, halide_type_of<int32_t>()));
+            auto temp1 = _Convert(inputs[0], NCHW);
+            auto temp2 = _Convert(backwardOutput[0], NCHW);
+            auto shape = _Shape(temp1);
+            auto temp3 = _Reshape(temp2, shape);
+            result[0] = _Convert(temp3, NC4HW4);
+        }
+        return result;
+    }
+};
+
+static void _create() {
+    static ReshapeGrad _c;
+    OpGrad::insert(OpType_Reshape, &_c);
+    OpGrad::insert(OpType_Squeeze, &_c);
+    OpGrad::insert(OpType_Unsqueeze, &_c);
+
+}
+
+REGISTER_GRAD(ReshapeGrad_cpp, _create);
+};
+