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,79 @@
+//
+//  CPUUnique.cpp
+//  MNN
+//
+//  Created by MNN on 2019/6/11.
+//  Copyright © 2018 Alibaba. All rights reserved.
+//
+
+#include <map>
+#include "backend/cpu/CPUUnique.hpp"
+namespace MNN {
+
+ErrorCode CPUUnique::onExecute(const std::vector<Tensor*>& inputs, const std::vector<Tensor*>& outputs) {
+    auto input = inputs[0];
+    auto output = outputs[0];
+    auto outputPtr = output->host<int32_t>();
+    int outputSize = 0;
+    auto eleSize = input->elementSize();
+    if (outputs.size() <= 2) {
+        std::map<int, int> idx_map;
+        for (int i = 0; i < eleSize; ++i) {
+            auto value = input->host<int32_t>()[i];
+            idx_map[value];
+        }
+        for (auto& kv : idx_map) {
+            kv.second = outputSize;
+            outputPtr[outputSize++] = kv.first;
+        }
+        if (outputs.size() > 1) {
+            auto outIdx = outputs[1]->host<int>();
+            for (int i = 0; i < eleSize; ++i) {
+                auto value = input->host<int32_t>()[i];
+                outIdx[i] = idx_map[value];
+            }
+        }
+    } else {
+        MNN_ASSERT(outputs.size() <= 4);
+        auto outIdx = outputs[1]->host<int>();
+        auto reverseIdx = outputs[2]->host<int>();
+        int* count = nullptr;
+        if (outputs.size() > 3) {
+            count = outputs[3]->host<int>();
+            ::memset(count, 0, outputs[3]->usize());
+        }
+        std::map<int, int> idx_map;
+        for (int i = 0; i < eleSize; ++i) {
+            auto value = input->host<int32_t>()[i];
+            idx_map[value];
+        }
+        for (auto& kv : idx_map) {
+            outIdx[outputSize] = -1;
+            kv.second = outputSize;
+            outputPtr[outputSize++] = kv.first;
+        }
+        for (int i = 0; i < eleSize; ++i) {
+            auto value = input->host<int32_t>()[i];
+            int pos = idx_map[value];
+            reverseIdx[i] = pos;
+            if (outIdx[pos] < 0) {
+                outIdx[pos] = i;
+            }
+            if (nullptr != count) {
+                count[pos]++;
+            }
+        }
+    }
+    return NO_ERROR;
+}
+class CPUUniqueCreator : public CPUBackend::Creator {
+public:
+    virtual Execution* onCreate(const std::vector<Tensor*>& inputs, const std::vector<Tensor*>& outputs,
+                                const MNN::Op* op, Backend* backend) const {
+        return new CPUUnique(backend);
+    }
+};
+
+REGISTER_CPU_OP_CREATOR(CPUUniqueCreator, OpType_Unique);
+
+}; // namespace MNN