read_onnx.cpp 274 Bytes
Newer Older
Paul's avatar
Paul committed
1

Paul's avatar
Paul committed
2
#include <migraphx/onnx.hpp>
Paul's avatar
Paul committed
3

Paul's avatar
Paul committed
4
int main(int argc, char const* argv[])
Paul's avatar
Paul committed
5
6
7
8
{
    if(argc > 1)
    {
        std::string file = argv[1];
Paul's avatar
Paul committed
9
        auto prog        = migraphx::parse_onnx(file);
Khalique's avatar
Khalique committed
10
        // std::cout << prog << std::endl;
11
        prog.print_graph(std::cout);
Paul's avatar
Paul committed
12
13
    }
}