Commit 6d5a34d2 authored by Paul's avatar Paul
Browse files

Format

parent 53a8997f
...@@ -63,13 +63,13 @@ template <index_int Axis, class Start, class InputPack, class F, class... Ts> ...@@ -63,13 +63,13 @@ template <index_int Axis, class Start, class InputPack, class F, class... Ts>
__device__ auto concat_each(index idx, Start start, InputPack input_pack, F f, Ts... ts) __device__ auto concat_each(index idx, Start start, InputPack input_pack, F f, Ts... ts)
{ {
return input_pack([&](auto g, auto x, auto... xs) { return input_pack([&](auto g, auto x, auto... xs) {
return concat_slices<Axis>(x, start, ts...)([&](auto z, auto... ys) { return concat_slices<Axis>(x, start, ts...)([&](auto z, auto... ys) {
idx.global_stride(x.get_shape().elements(), idx.global_stride(x.get_shape().elements(),
[&](auto i) { z[i] = f(g(x[i], xs[i]...), ys[i]...); }); [&](auto i) { z[i] = f(g(x[i], xs[i]...), ys[i]...); });
return start + concat_ends<Axis>(x); return start + concat_ends<Axis>(x);
});
}); });
});
} }
template <index_int Axis, class... InputPacks> template <index_int Axis, class... InputPacks>
......
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