Function _replaceConstOp

tools/cpp/compilefornpu.cpp:498–542

}

static NetT* _replaceConstOp(const void* buffer, size_t bufferSize,
                             std::map<int, std::tuple<int, int, std::vector<int>, std::vector<char>>>& constTensorData,
                             std::vector<int>& constOpId) {
    auto net = flatbuffers::GetRoot<Net>(buffer)->UnPack();
    for (int i = 0; i < constOpId.size(); ++i) {
        auto op = net->oplists[constOpId[i]].get();
        auto index = op->outputIndexes[0];
        auto iter = constTensorData.find(index);
        if (iter == constTensorData.end()) {
            continue;
        }
        auto name = op->name;
        std::unique_ptr<MNN::OpT> newOp(new OpT);
        newOp->type = OpType_Const;
        newOp->name = name + "_Const";
        auto blob = new BlobT;
        blob->dataFormat = (MNN_DATA_FORMAT)std::get<0>(iter->second);
        blob->dims = std::get<2>(iter->second);
        if (std::get<1>(iter->second) == halide_type_float) {
            blob->dataType = DataType_DT_FLOAT;
            blob->float32s.resize(std::get<3>(iter->second).size() / 4);
            memcpy(blob->float32s.data(), std::get<3>(iter->second).data(), std::get<3>(iter->second).size());
        } else if (std::get<1>(iter->second) == halide_type_int) {
            blob->dataType = DataType_DT_INT32;
            blob->int32s.resize(std::get<3>(iter->second).size() / 4);
            memcpy(blob->int32s.data(), std::get<3>(iter->second).data(), std::get<3>(iter->second).size());
        } else {
            continue;
        }
        newOp->main.value = blob;
        newOp->main.type = OpParameter_Blob;
        newOp->outputIndexes = {index};
        net->oplists[constOpId[i]] = std::move(newOp);
    }
    auto oplist = std::move(net->oplists);
    for (auto& op : oplist) {
        if (nullptr != op.get()) {
            net->oplists.emplace_back(std::move(op));
        }
    }
    return net;
    // flatbuffers::GetRoot<Net>(buffer)->Pack(builder, net);
    // std::ofstream outputOs(dstMNN, std::ios::binary);
    // outputOs.write((const char*)builder.GetBufferPointer(), builder.GetSize());
}

static std::vector<int> _findBreakIndex(const SubModuleInfo& info, const Net* net, std::shared_ptr<Schedule::ScheduleInfo> sharedConst) {
    // 0: not used, 1: const, 2: output