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

Formatting

parent f199ea9e
......@@ -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&& s = x.get_shape();
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");
return [&](auto v) {
s.visit_type([&](auto as) {
......
......@@ -27,10 +27,8 @@ struct cpu_convolution
auto wei_h = weights.get_shape().lens()[2];
auto wei_w = weights.get_shape().lens()[3];
dfor(in_n,
in_c,
in_h,
in_w)([&](std::size_t o, std::size_t w, std::size_t i, std::size_t j) {
dfor(in_n, in_c, 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_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