Commit 224cb7cf authored by wsttiger's avatar wsttiger
Browse files

Formatting

parent 73ca4689
...@@ -309,10 +309,12 @@ struct cpu_concat ...@@ -309,10 +309,12 @@ struct cpu_concat
auto argl = args[l]; auto argl = args[l];
std::size_t nelements = argl.get_shape().elements(); std::size_t nelements = argl.get_shape().elements();
visit_all(result, argl)([&](auto output, auto input) { visit_all(result, argl)([&](auto output, auto input) {
auto slice_shape = shape{output_shape.type(), input.get_shape().lens(), output_shape.strides()}; auto slice_shape =
auto slice = make_view(slice_shape, output.data()+coffsets[l]); shape{output_shape.type(), input.get_shape().lens(), output_shape.strides()};
for(std::size_t i = 0; i < nelements; i++) { auto slice = make_view(slice_shape, output.data() + coffsets[l]);
slice[i] = input[i]; for(std::size_t i = 0; i < nelements; i++)
{
slice[i] = input[i];
} }
}); });
} }
......
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