"src/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "c722117d89a851915d46edb828d2ff0654849a04"
read_tf.cpp 375 Bytes
Newer Older
Khalique's avatar
Khalique committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#include <migraphx/tf.hpp>

int main(int argc, char const* argv[])
{
    if(argc > 1)
    {
        bool is_nhwc = true;
        if(argc > 2)
        {
            if(argv[2] == "nchw")
                is_nhwc = false;
        }
        std::string file = argv[1];
        auto prog        = migraphx::parse_tf(file, is_nhwc);
        std::cout << prog << std::endl;
    }
}