Skip to content

GitLab

Commit 035a04eb authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

formatting

parent b0d03793
Show whitespace changes
Inline Side-by-side
Showing with 9 additions and 6 deletions
src/rewrite_gemm.cpp
View file @ 035a04eb
...@@ -49,12 +49,15 @@ void rewrite_gemm::apply(module& m) const ...@@ -49,12 +49,15 @@ void rewrite_gemm::apply(module& m) const
continue; continue;
auto in1 = inputs.at(1); auto in1 = inputs.at(1);
auto in0_transposed = m.insert_instruction(ins, make_op("transpose", {{"permutation", {0,3,1,2}}}), in0); auto in0_transposed =
auto in1_transposed = m.insert_instruction(ins, make_op("transpose", {{"permutation", {3,2,1,0}}}), in1); m.insert_instruction(ins, make_op("transpose", {{"permutation", {0, 3, 1, 2}}}), in0);
auto in1_transposed =
m.insert_instruction(ins, make_op("transpose", {{"permutation", {3, 2, 1, 0}}}), in1);
auto conv = make_op("convolution"); auto conv = make_op("convolution");
auto conv_out = m.replace_instruction(ins, conv, {in0_transposed, in1_transposed}); auto conv_out = m.replace_instruction(ins, conv, {in0_transposed, in1_transposed});
auto conv_transpose = m.add_instruction(make_op("transpose", {{"permutation", {0,2,3,1}}}), conv_out); auto conv_transpose =
m.add_instruction(make_op("transpose", {{"permutation", {0, 2, 3, 1}}}), conv_out);
// m.insert_instruction(std::next(conv_transpose), make_op("unsqueeze")); // m.insert_instruction(std::next(conv_transpose), make_op("unsqueeze"));
} }
} }
......
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