Function channelPruneConvert

tools/converter/source/common/ChannelPruneConvert.cpp:291–366

}

void channelPruneConvert(std::unique_ptr<MNN::NetT>& netT, MNN::Compression::Pipeline proto) {
    bool filterPruned = false;
    for (const auto& algo : proto.algo()) {
        if (algo.type() == Compression::CompressionAlgo::PRUNE) {
            auto prune_type = algo.prune_params().type();
            auto prune_algo_type = MNN::SparseAlgo(prune_type);
            if (prune_type == Compression::PruneParams_PruneType_FILTER) {
                filterPruned = true;
                break;
            }
        }
    }
    
    if (!filterPruned) {
        return;
    }
 
    std::map<std::string, TensorMaskInfo> netMaskInfo;
    for (auto tensorName : netT->tensorName) {
        netMaskInfo[tensorName] = TensorMaskInfo();
    }

    std::set<std::string> notSafeConvNames;
    for (auto& op : netT->oplists) {
        analyzePruneInfo(op, netT, nullptr, netMaskInfo, notSafeConvNames);
    }

    std::set<std::string>::iterator iter;
    if (!notSafeConvNames.empty()) {
        for (auto& info : netMaskInfo) {
            iter = std::find(notSafeConvNames.begin(), notSafeConvNames.end(), info.second.oriConvName);
            if (iter != notSafeConvNames.end()) {
                for (int i = 0; i < info.second.mask.size(); i++) {
                    if (info.second.mask[i] == 0) {
                        info.second.mask[i] = 1;
                    }
                }
            }
        }
    }

    for (auto& op : netT->oplists) {
        channelPrune(op, netT, nullptr, netMaskInfo);
    }


    for (auto& subgraph : netT->subgraphs) {
        std::map<std::string, TensorMaskInfo> subgraphMaskInfo;
        for (auto tensorName : subgraph->tensors) {
            subgraphMaskInfo[tensorName] = TensorMaskInfo();
        }

        std::set<std::string> notSafeConvNames;
        for (auto& op : subgraph->nodes) {
            analyzePruneInfo(op, netT, subgraph.get(), subgraphMaskInfo, notSafeConvNames);
        }

        std::set<std::string>::iterator iter;