Skip to content

GitLab

Commit 9ac6a4a8 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 08207d66
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
src/rewrite_gru.cpp
View file @ 9ac6a4a8
...@@ -103,14 +103,16 @@ void rewrite_gru::apply(program& prog) const ...@@ -103,14 +103,16 @@ void rewrite_gru::apply(program& prog) const
gru_op.actv_funcs.at(2), gru_op.actv_funcs.at(2),
gru_op.actv_funcs.at(3)); gru_op.actv_funcs.at(3));
auto final_output = prog.insert_instruction(ins, op::concat{0}, ret_forward[1], ret_reverse[1]); auto final_output =
prog.insert_instruction(ins, op::concat{0}, ret_forward[1], ret_reverse[1]);
// add the dimension of num_direction // add the dimension of num_direction
ret_forward[0] = prog.insert_instruction(ins, op::unsqueeze{{1}}, ret_forward[0]); ret_forward[0] = prog.insert_instruction(ins, op::unsqueeze{{1}}, ret_forward[0]);
ret_reverse[0] = prog.insert_instruction(ins, op::unsqueeze{{1}}, ret_reverse[0]); ret_reverse[0] = prog.insert_instruction(ins, op::unsqueeze{{1}}, ret_reverse[0]);
// concat the forward and reverse output // concat the forward and reverse output
auto replaced_arg = prog.replace_instruction(ins, op::concat{1}, {ret_forward[0], ret_reverse[0]}); auto replaced_arg =
prog.replace_instruction(ins, op::concat{1}, {ret_forward[0], ret_reverse[0]});
replaced_arg->add_output(final_output); replaced_arg->add_output(final_output);
} }
else else
......
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