Function _needSplitNet

tools/cpp/compilefornpu.cpp:631–641

}

static bool _needSplitNet(const Net* net, const std::set<int>& inputIndexes, const std::set<int>& outputIndexes) {
    auto selectOps = _collectNeededOps(net, inputIndexes, outputIndexes);
    for (int si = 0; si < selectOps.size(); ++si) {
        auto i = selectOps[si];
        auto op = net->oplists()->GetAs<Op>(i);
        if (isBreakOp(op) || gExtraBreakOpIndexes.count(i) > 0) {
            return true;
        }
    }
    return false;
}

static std::vector<SubModuleInfo> _createSubModuleInfo(const Net* net, const std::set<int>& inputIndexes, const std::set<int>& outputIndexes, const std::set<int>& noComputeIndexes, std::shared_ptr<Schedule::ScheduleInfo> sharedConst) {
    std::vector<SubModuleInfo> submodule;