Commit 228b665c authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

formatting

parent 98b8dff1
...@@ -66,14 +66,16 @@ void transform_convolutions(module& m, bool skip_elim_contiguous) ...@@ -66,14 +66,16 @@ void transform_convolutions(module& m, bool skip_elim_contiguous)
if(args[i]->name() != "layout" and args[i]->get_shape().standard()) if(args[i]->name() != "layout" and args[i]->get_shape().standard())
{ {
// std::cout << "HERE2" << std::endl; // std::cout << "HERE2" << std::endl;
args[i] = m.insert_instruction(ins, make_op("layout", {{"permutation", {0, 2, 3, 1}}}), args[i]); args[i] = m.insert_instruction(
ins, make_op("layout", {{"permutation", {0, 2, 3, 1}}}), args[i]);
// m.debug_print(args); // m.debug_print(args);
} }
} }
} }
else else
std::transform(args.begin(), args.end(), args.begin(), [&](auto& i) { std::transform(args.begin(), args.end(), args.begin(), [&](auto& i) {
return m.insert_instruction(ins, make_op("layout", {{"permutation", {0, 2, 3, 1}}}), i); return m.insert_instruction(
ins, make_op("layout", {{"permutation", {0, 2, 3, 1}}}), i);
}); });
auto conv = m.insert_instruction(ins, ins->get_operator(), args); auto conv = m.insert_instruction(ins, ins->get_operator(), args);
auto c = conv; auto c = conv;
......
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