"src/nni_manager/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "1fec96c90c579c5e54e5b0eebac374e24084aa6d"
Commit 8ba66ebc authored by Paul's avatar Paul
Browse files

Format

parent 7fa93728
...@@ -81,15 +81,15 @@ __device__ auto concat2(InputPacks... input_packs) ...@@ -81,15 +81,15 @@ __device__ auto concat2(InputPacks... input_packs)
auto idx = make_index(); auto idx = make_index();
fold([&](auto start, auto input_pack) { fold([&](auto start, auto input_pack) {
return input_pack([&](auto g, auto x, auto... xs) { return input_pack([&](auto g, auto x, auto... xs) {
concat_slices<Axis>(x, start, ts...)([&](auto z, auto... ys) { 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);
}); });
}); });
})(_c<0>, input_packs...); })(_c<0>, input_packs...);
} }
} // namespace migraphx } // namespace migraphx
#endif // MIGRAPHX_GUARD_KERNELS_CONCAT_HPP #endif // MIGRAPHX_GUARD_KERNELS_CONCAT_HPP
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