onnx.hpp 362 Bytes
Newer Older
Paul's avatar
Paul committed
1
2
#ifndef MIGRAPHX_GUARD_MIGRAPHLIB_ONNX_HPP
#define MIGRAPHX_GUARD_MIGRAPHLIB_ONNX_HPP
Paul's avatar
Paul committed
3

Paul's avatar
Paul committed
4
5
#include <migraphx/program.hpp>
#include <migraphx/config.hpp>
Paul's avatar
Paul committed
6

Paul's avatar
Paul committed
7
namespace migraphx {
Paul's avatar
Paul committed
8
inline namespace MIGRAPHX_INLINE_NS {
Paul's avatar
Paul committed
9

Paul's avatar
Paul committed
10
/// Create a program from an onnx file
Paul's avatar
Paul committed
11
12
program parse_onnx(const std::string& name);

Paul's avatar
Paul committed
13
} // namespace MIGRAPHX_INLINE_NS
Paul's avatar
Paul committed
14
} // namespace migraphx
Paul's avatar
Paul committed
15
16

#endif