Function initGetMaxValue

tools/cpp/getLinearInput.hpp:187–238

}

inline void initGetMaxValue(const std::string& maxValueFileName) {
    initMaxValueFile(maxValueFileName);
    
    static bool registered = false;
    if (!registered) {
        std::atexit(cleanupAtExit);
        registered = true;
    }

    MNN::TensorCallBackWithInfo beforeCallBack = [](const std::vector<MNN::Tensor*>& ntensors, const MNN::OperatorInfo* info) {
        auto opName = info->name();
        if (info->type() == "Copy") {
            return true;
        }
        if (opName.find("Linear") == std::string::npos || opName.find("raster") != std::string::npos) {
            return true;
        }

        for (int i = 0; i < ntensors.size(); ++i) {
            auto ntensor = ntensors[i];
            auto outDimType = ntensor->getDimensionType();
            std::shared_ptr<MNN::Tensor> expectTensor(new MNN::Tensor(ntensor, outDimType));
            bool res = ntensor->copyToHostTensor(expectTensor.get());
            if (res) {
                ntensor = expectTensor.get();
            }
            {
                auto ninput = MNN::Express::Variable::create(MNN::Express::Expr::create(ntensor));
                if (nullptr == ninput->getInfo()) {
                    MNN_ERROR("Alloc memory or compute size error\n");
                    return false;
                }
                ninput = MNN::Express::_Convert(ninput, MNN::Express::NHWC);
                ninput = MNN::Express::_Abs(ninput);  
                ninput = MNN::Express::_Reshape(ninput, {-1}); 

                auto kv = MNN::Express::_TopKV2(ninput, MNN::Express::_Scalar<int>(1));     
                auto maxValues = kv[0];  
                auto maxValueFloat = maxValues->readMap<float>()[0];

                writeMaxValueRealtime(opName, maxValueFloat);
            }
        }
        return true;
    };

    MNN::TensorCallBackWithInfo callBack = [](const std::vector<MNN::Tensor*>& ntensors, const MNN::OperatorInfo* info) {
        return true;
    };

    MNN::Express::ExecutorScope::Current()->setCallBack(std::move(beforeCallBack), std::move(callBack));
}

inline void closeAllFiles() {
    closeThresholdFile();