Commit 837bce78 authored by Paul's avatar Paul
Browse files

Formatting

parent f199ea9e
...@@ -96,12 +96,12 @@ struct raw_data ...@@ -96,12 +96,12 @@ struct raw_data
} }
}; };
template<class T, class... Ts> template <class T, class... Ts>
auto visit_all(T&& x, Ts&&... xs) auto visit_all(T&& x, Ts&&... xs)
{ {
auto&& s = x.get_shape(); auto&& s = x.get_shape();
std::initializer_list<shape::type_t> types = {xs.get_shape().type()...}; std::initializer_list<shape::type_t> types = {xs.get_shape().type()...};
if (!std::all_of(types.begin(), types.end(), [&](shape::type_t t) { return t == s.type(); })) if(!std::all_of(types.begin(), types.end(), [&](shape::type_t t) { return t == s.type(); }))
RTG_THROW("Types must be the same"); RTG_THROW("Types must be the same");
return [&](auto v) { return [&](auto v) {
s.visit_type([&](auto as) { s.visit_type([&](auto as) {
......
...@@ -27,10 +27,8 @@ struct cpu_convolution ...@@ -27,10 +27,8 @@ struct cpu_convolution
auto wei_h = weights.get_shape().lens()[2]; auto wei_h = weights.get_shape().lens()[2];
auto wei_w = weights.get_shape().lens()[3]; auto wei_w = weights.get_shape().lens()[3];
dfor(in_n, dfor(in_n, in_c, in_h, in_w)(
in_c, [&](std::size_t o, std::size_t w, std::size_t i, std::size_t j) {
in_h,
in_w)([&](std::size_t o, std::size_t w, std::size_t i, std::size_t j) {
const int start_x = i * op.stride[0] - op.padding[0]; const int start_x = i * op.stride[0] - op.padding[0];
const int start_y = j * op.stride[1] - op.padding[1]; const int start_y = j * op.stride[1] - op.padding[1];
......
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