Method SaveModel

serializer/tengine/tm_serializer.cpp:45–79

extern bool register_tm2_serializer();

bool TmSerializer::SaveModel(const std::vector<std::string>& file_list, Graph* graph)
{
    /* Check the file number */
    if(file_list.size() != GetFileNum())
        return false;

    /* Open the tengine model file */
    int fd = open(file_list[0].c_str(), O_RDWR | O_CREAT | O_TRUNC, 0666);
    if(fd == -1)
    {
        LOG_ERROR() << "Could not open " << file_list[0] << "\n";
        return false;
    }

    std::vector<void*> addr_list;
    std::vector<int> size_list;

    if(!SaveModel(addr_list, size_list, graph))
    {
        close(fd);
        return false;
    }

    void* buf = addr_list[0];
    int size = size_list[0];
    int ret = write(fd, buf, size);

    close(fd);
    free(buf);

    if(ret != size)
        return false;
    else
        return true;
}

bool TmSerializer::SaveModel(std::vector<void*>& addr_list, std::vector<int>& size_list, Graph* graph)
{