Skip to content

GitLab

Commit d4e9ae1c authored by Shucai Xiao's avatar Shucai Xiao
Browse files

code backup

parent 6d8fcb3d
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
src/rewrite_rnn.cpp
View file @ d4e9ae1c
...@@ -789,6 +789,11 @@ void rewrite_rnn::apply_lstm(program& prog, instruction_ref ins) const ...@@ -789,6 +789,11 @@ void rewrite_rnn::apply_lstm(program& prog, instruction_ref ins) const
} }
else else
{ {
ret_forward[0] =
prog.insert_instruction(ins, op::concat{0}, ret_forward[0], ret_forward[1]);
ret_reverse[0] =
prog.insert_instruction(ins, op::concat{0}, ret_reverse[1], ret_reverse[0]);
prog.replace_instruction(ins, op::concat{1}, {ret_forward[0], ret_reverse[0]});
} }
} }
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