"vscode:/vscode.git/clone" did not exist on "d047d6f4d869cf8a227280ae51577eb175f67d6a"
Commit 228b665c authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

formatting

parent 98b8dff1
...@@ -61,24 +61,26 @@ void transform_convolutions(module& m, bool skip_elim_contiguous) ...@@ -61,24 +61,26 @@ void transform_convolutions(module& m, bool skip_elim_contiguous)
// std::cout << "HERE" << std::endl; // std::cout << "HERE" << std::endl;
for(auto i = 0; i < args.size(); i++) for(auto i = 0; i < args.size(); i++)
{ {
// std::cout << args[i]->name() << std::endl; // std::cout << args[i]->name() << std::endl;
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;
if(not skip_elim_contiguous) if(not skip_elim_contiguous)
c = m.insert_instruction(ins, make_op("contiguous"), conv); c = m.insert_instruction(ins, make_op("contiguous"), conv);
m.replace_instruction(ins, c); m.replace_instruction(ins, c);
} }
} }
......
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