wehub-resource-sync
36 lines
1.1 KiB
C++
36 lines
1.1 KiB
C++
//
|
|
// RemoveOutputTensorConvert.cpp
|
|
// MNNConverter
|
|
//
|
|
// Created by MNN on 2020/02/18.
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
|
//
|
|
|
|
#include "../PostTreatUtils.hpp"
|
|
#include "../Global.hpp"
|
|
#include "config.hpp"
|
|
using namespace MNN;
|
|
class RemoveOutputTensorConvert : public PostConverter {
|
|
public:
|
|
virtual bool onExecute(std::unique_ptr<MNN::NetT>& net) const override {
|
|
auto config = Global<modelConfig>::Get();
|
|
if (config->keepInputFormat) {
|
|
return true;
|
|
}
|
|
for (auto iter = net->oplists.begin(); iter != net->oplists.end();) {
|
|
auto& op = *iter;
|
|
if (op->outputIndexes.empty() || op->type != OpType_ConvertTensor) {
|
|
iter++;
|
|
continue;
|
|
}
|
|
if (PostTreatUtils::_findOpByInputIndex(op->outputIndexes[0], net.get()).size() > 0) {
|
|
iter++;
|
|
continue;
|
|
}
|
|
iter = net->oplists.erase(iter);
|
|
}
|
|
return true;
|
|
}
|
|
};
|
|
static PostConverterRegister<RemoveOutputTensorConvert> __l("RemoveOutputTensorConvert");
|