Function writeFb

tools/converter/source/common/writeFb.cpp:177–321

    return 0;
}
int writeFb(std::unique_ptr<MNN::NetT>& netT, const modelConfig& config, std::unique_ptr<MNN::OpT>&& metaOp) {
    postTreat(netT, config);
    // Merge Meta to metaOp
    auto oplist = std::move(netT->oplists);
    for (auto& op : oplist) {
        if (op->type == OpType_Extra) {
            auto dstExtra = metaOp->main.AsExtra();
            auto extra = op->main.AsExtra();
            if (extra->type == "Meta" && extra->engine == "MNN") {
                for (auto& attr : extra->attr) {
                    dstExtra->attr.emplace_back(std::move(attr));
                }
                // Remove meta op
                continue;
            }
        }
        netT->oplists.emplace_back(std::move(op));
    }
    std::set<std::string> notSupportOps;
    
    // Detect unsupport op
    auto CheckIfNotSupported = [&] (const std::unique_ptr<MNN::OpT>& op) {
        if (op->type == MNN::OpType_Extra) {
            if (op->main.AsExtra()->engine != "MNN") {
                notSupportOps.insert(op->main.AsExtra()->engine + "::" + op->main.AsExtra()->type);
            }
        }
    };
    for (auto& op : netT->oplists) {
        CheckIfNotSupported(op);
    }
    for (auto& subgraph : netT->subgraphs) {
        for (auto& op : subgraph->nodes) {
            CheckIfNotSupported(op);
        }
    }

    std::ostringstream notSupportInfo;
    if (!notSupportOps.empty() && !config.allowCustomOp) {
        for (auto name : notSupportOps) {
            notSupportInfo << name << " | ";
        }
        auto opNames = notSupportInfo.str();
        LOG(FATAL) << "These Op Not Support: " << opNames.substr(0, opNames.size() - 2);
        return 1;
    }

    // dump input and output tensor name
    {
        std::set<int> inputIdx, outputIdx, realOutput;
        std::vector<int> realInput;
        for (const auto& op : netT->oplists) {
            for (auto i : op->inputIndexes) {
                inputIdx.insert(i);
            }
            for (auto o : op->outputIndexes) {
                outputIdx.insert(o);
                if (op->type == OpType_Input) {