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

Formatting

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