"vscode:/vscode.git/clone" did not exist on "224b7aa0da7401051f5127de8b840c7fb4225bfd"
Commit 035a04eb authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

formatting

parent b0d03793
......@@ -49,12 +49,15 @@ void rewrite_gemm::apply(module& m) const
continue;
auto in1 = inputs.at(1);
auto in0_transposed = 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 in0_transposed =
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_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"));
}
}
......
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