Function InitModelConfig

tools/quantization/quantize.cc:59–97

}

int InitModelConfig(ModelConfig& model_config, std::string proto_file, std::string model_file) {
    FILE* fp = fopen(proto_file.c_str(), "r");
    if (fp == NULL) {
        printf("invalid proto file\n");
        return -1;
    }

    fseek(fp, 0, SEEK_END);
    int size = ftell(fp);
    rewind(fp);
    char* buffer = new char[size + 1];
    memset(buffer, 0, size + 1);
    int ret = fread(buffer, 1, size, fp);
    if (ret != size) {
        printf("read proto_file failed!\n");
        return -1;
    }
    fclose(fp);

    ParseProtoFile(buffer, size);
    std::string buffer_str(buffer);

    model_config.params.push_back(buffer_str);
    delete[] buffer;

    {
        std::ifstream model_stream(model_file);
        if (!model_stream.is_open() || !model_stream.good()) {
            printf("read model_file failed!\n");
            return -1;
        }
        std::string model_content =
            std::string((std::istreambuf_iterator<char>(model_stream)), std::istreambuf_iterator<char>());

        model_config.params.push_back(model_content);
    }

    return 0;
}

bool GetInputType(std::string name, FileFormat& format) {
    int pos = name.rfind('.');