harness.cpp

#include <armnnOnnxParser/IOnnxParser.hpp>

#include <exception>
#include <iostream>

int main(int argc, char** argv)
{
    if (argc != 2)
    {
        std::cerr << "usage: " << argv[0] << " model.onnx\n";
        return 2;
    }

    try
    {
        auto parser = armnnOnnxParser::IOnnxParser::Create();
        auto network = parser->CreateNetworkFromBinaryFile(argv[1]);
        if (!network)
        {
            std::cerr << "parser returned a null network\n";
            return 3;
        }
        std::cout << "parsed successfully\n";
        return 0;
    }
    catch (const std::exception& error)
    {
        std::cerr << "parser exception: " << error.what() << '\n';
        return 1;
    }
}