Function _splitSubModuleForShapeConst

tools/cpp/compilefornpu.cpp:597–629

    return res;
}
static std::vector<SubModuleInfo> _splitSubModuleForShapeConst(const std::vector<SubModuleInfo>& origin, const Net* net, std::shared_ptr<Schedule::ScheduleInfo> sharedConst) {
    std::vector<SubModuleInfo> res;
    for (auto& m : origin) {
        if (m.isBreak) {
            res.emplace_back(std::move(m));
            continue;
        }
        auto breakIndexes = _findBreakIndex(m, net, sharedConst);
        if (breakIndexes.size() > 0) {
            int current = 0;
            for (auto breakIndex : breakIndexes) {
                // Split
                if (breakIndex > current) {
                    SubModuleInfo m0;
                    m0.opList.insert(m0.opList.begin(), m.opList.begin() + current, m.opList.begin() + breakIndex);
                    res.emplace_back(std::move(m0));
                }
                SubModuleInfo m1;
                m1.opList = {m.opList[breakIndex]};
                res.emplace_back(std::move(m1));
                current = breakIndex + 1;
            }
            if (current < m.opList.size()) {
                SubModuleInfo m2;
                m2.opList.insert(m2.opList.begin(), m.opList.begin() + current, m.opList.end());
                res.emplace_back(std::move(m2));
            }
        } else {
            res.emplace_back(std::move(m));
        }
    }
    return res;
}

static bool _needSplitNet(const Net* net, const std::set<int>& inputIndexes, const std::set<int>& outputIndexes) {
    auto selectOps = _collectNeededOps(net, inputIndexes, outputIndexes);