Method load

express/module/PipelineModule.cpp:677–714

}

Module* PipelineModule::load(const std::vector<std::string>& inputs, const std::vector<std::string>& outputs, const uint8_t* buffer, size_t length, const std::shared_ptr<MNN::Express::Executor::RuntimeManager> rtMgr, const Module::Config* config) {
    // Create Subgraph
    auto net = GetNet(buffer);
    if (nullptr == net->oplists() || nullptr == net->tensorName()) {
        MNN_ERROR("Invalid net, for null oplist or tensorName\n");
        return nullptr;
    }
    Module::Config defaultConfig;
    if (nullptr == config) {
        config = &defaultConfig;
    }
    auto subGraphs = net->subgraphs();
    if (config->dynamic) {
        // TODO: Support subgraph
        if (nullptr == subGraphs) {
            auto varMap = MNN::Express::Variable::loadMap(buffer, length);
            std::vector<MNN::Express::VARP> inputsVar(inputs.size());
            for (int i=0; i<inputs.size(); ++i) {
                inputsVar[i] = varMap[inputs[i]];
            }
            std::vector<MNN::Express::VARP> outputsVar(outputs.size());
            for (int i=0; i<outputs.size(); ++i) {
                outputsVar[i] = varMap[outputs[i]];
            }
            return extract(inputsVar, outputsVar, false);
        } else {
            MNN_ERROR("Don't support subgraph for dynamic load, turn back to static load\n");
        }
    }
    std::map<std::string, SubGraph> subGraphMap;
    _createSubGraph(net, rtMgr, config, subGraphMap);
    std::shared_ptr<BufferStorage> bufferStorage(new BufferStorage);
    bufferStorage->storage = new uint8_t[length];
    ::memcpy(bufferStorage->storage, buffer, length);
    bufferStorage->offset = 0;
    bufferStorage->allocated_size = length;
    return load(inputs, outputs, bufferStorage, rtMgr, config, subGraphMap);
}

Module* PipelineModule::load(const std::vector<std::string>& inputs, const std::vector<std::string>& outputs, std::shared_ptr<BufferStorage> bufferStorage, std::shared_ptr<MNN::Express::Executor::RuntimeManager> rtMgr, const Module::Config* config, std::map<std::string, SubGraph>& subGraphMap) {
    MNN_ASSERT(nullptr != rtMgr);