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

Format

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