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,66 @@
+//
+//  ResolveTfShapeN.cpp
+//  MNNConverter
+//
+//  Created by MNN on 2019/11/27.
+//  Copyright © 2018, Alibaba Group Holding Limited
+//
+
+#include "../PostTreatUtils.hpp"
+#include "flatbuffers/util.h"
+
+class ResolveTfShapeN : public PostConverter {
+public:
+    virtual bool onExecute(std::unique_ptr<MNN::NetT>& net) const override {
+        if (net->sourceType != MNN::NetSource_TENSORFLOW) {
+            return true;
+        }
+
+        std::set<MNN::OpT*> readyToDelete;
+
+        for (auto iter = net->oplists.begin(); iter != net->oplists.end();) {
+            const auto op = iter->get();
+            if (op->type != MNN::OpType_Extra) {
+                iter++;
+                continue;
+            }
+            auto attr    = op->main.AsExtra();
+            auto& optype = attr->type;
+            if (optype != "ShapeN") {
+                iter++;
+                continue;
+            }
+            int shapeNumber    = 1;
+            const int attrSize = attr->attr.size();
+            for (int k = 0; k < attrSize; ++k) {
+                auto& key = attr->attr[k]->key;
+                if (key == "N") {
+                    shapeNumber = attr->attr[k]->i;
+                }
+            }
+            readyToDelete.insert(op);
+            // expand ShapeN to N Shapes
+            // insert N Shape before ShapeN, then delete the shapeN
+            for (int i = 0; i < shapeNumber; ++i) {
+                std::unique_ptr<MNN::OpT> curShape(new MNN::OpT);
+                curShape->name       = op->name + flatbuffers::NumToString(i);
+                curShape->type       = MNN::OpType_Shape;
+                curShape->main.value = nullptr;
+                curShape->inputIndexes.push_back(op->inputIndexes[i]);
+                curShape->outputIndexes.push_back(op->outputIndexes[i]);
+                iter = net->oplists.insert(iter, std::move(curShape));
+                iter++;
+            }
+
+            iter++;
+        }
+
+        for (auto op : readyToDelete) {
+            PostTreatUtils::_removeOpInNet(op, net.get());
+        }
+
+        return true;
+    }
+};
+
+static PostConverterRegister<ResolveTfShapeN> __shapen("ResolveTfShapeN");