Commit 072459db authored by Paul's avatar Paul
Browse files

Formatting

parent 3032998d
...@@ -124,10 +124,7 @@ struct tensor_view ...@@ -124,10 +124,7 @@ struct tensor_view
return m_data + this->size(); return m_data + this->size();
} }
std::vector<T> to_vector() const std::vector<T> to_vector() const { return std::vector<T>(this->begin(), this->end()); }
{
return std::vector<T>(this->begin(), this->end());
}
friend std::ostream& operator<<(std::ostream& os, const tensor_view<T>& x) friend std::ostream& operator<<(std::ostream& os, const tensor_view<T>& x)
{ {
......
...@@ -137,11 +137,10 @@ PYBIND11_MODULE(migraphx, m) ...@@ -137,11 +137,10 @@ PYBIND11_MODULE(migraphx, m)
new(&x) migraphx::argument(to_shape(info), info.ptr); new(&x) migraphx::argument(to_shape(info), info.ptr);
}) })
.def("get_shape", &migraphx::argument::get_shape) .def("get_shape", &migraphx::argument::get_shape)
.def("tolist", [](migraphx::argument& x) { .def("tolist",
[](migraphx::argument& x) {
py::list l{x.get_shape().elements()}; py::list l{x.get_shape().elements()};
visit(x, [&](auto data) { visit(x, [&](auto data) { l = py::cast(data.to_vector()); });
l = py::cast(data.to_vector());
});
return l; return l;
}) })
.def("__eq__", std::equal_to<migraphx::argument>{}) .def("__eq__", std::equal_to<migraphx::argument>{})
......
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