Commit 75aa4baf authored by Khalique's avatar Khalique
Browse files

formatting

parent ecf260d8
...@@ -7,8 +7,6 @@ ...@@ -7,8 +7,6 @@
namespace migraphx { namespace migraphx {
inline namespace MIGRAPHX_INLINE_NS { inline namespace MIGRAPHX_INLINE_NS {
/// Create a program from an onnx file /// Create a program from an onnx file
program parse_onnx(const std::string& name); program parse_onnx(const std::string& name);
......
...@@ -156,9 +156,13 @@ PYBIND11_MODULE(migraphx, m) ...@@ -156,9 +156,13 @@ PYBIND11_MODULE(migraphx, m)
.def("__ne__", std::not_equal_to<migraphx::program>{}) .def("__ne__", std::not_equal_to<migraphx::program>{})
.def("__repr__", [](const migraphx::program& p) { return migraphx::to_string(p); }); .def("__repr__", [](const migraphx::program& p) { return migraphx::to_string(p); });
// m.def("parse_onnx", &migraphx::parse_onnx); // m.def("parse_onnx", &migraphx::parse_onnx);
m.def("parse_tf", &migraphx::parse_tf, "Parse tf protobuf (default format is nhwc)", py::arg("filename"), py::arg("is_nhwc") = true); m.def("parse_tf",
&migraphx::parse_tf,
"Parse tf protobuf (default format is nhwc)",
py::arg("filename"),
py::arg("is_nhwc") = true);
m.def("get_target", [](const std::string& name) -> migraphx::target { m.def("get_target", [](const std::string& name) -> migraphx::target {
if(name == "cpu") if(name == "cpu")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment