Commit 18e0c3fb authored by Paul's avatar Paul
Browse files

Format

parent 9c6eab14
...@@ -1135,15 +1135,14 @@ TEST_CASE(transpose_unsqueeze_concat) ...@@ -1135,15 +1135,14 @@ TEST_CASE(transpose_unsqueeze_concat)
std::vector<migraphx::instruction_ref> unsqueezed_args; std::vector<migraphx::instruction_ref> unsqueezed_args;
int64_t axis = 3; int64_t axis = 3;
std::transform(args.begin(), std::transform(
args.begin(),
args.end(), args.end(),
std::back_inserter(unsqueezed_args), std::back_inserter(unsqueezed_args),
[&](migraphx::instruction_ref arg) { [&](migraphx::instruction_ref arg) {
return m1.add_instruction( return m1.add_instruction(migraphx::make_op("unsqueeze", {{"axes", {axis}}}), arg);
migraphx::make_op("unsqueeze", {{"axes", {axis}}}), arg);
}); });
m1.add_instruction(migraphx::make_op("concat", {{"axis", axis}}), m1.add_instruction(migraphx::make_op("concat", {{"axis", axis}}), unsqueezed_args);
unsqueezed_args);
} }
// TODO: This could be simplified to a single transpose after concat // TODO: This could be simplified to a single transpose after concat
migraphx::module m2 = m1; migraphx::module m2 = m1;
......
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