code backup

d4e9ae1c · Shucai Xiao · 6d8fcb3d · d4e9ae1c
Commit d4e9ae1c authored Feb 07, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

src/rewrite_rnn.cpp src/rewrite_rnn.cpp +5 -0

No files found.
--- a/src/rewrite_rnn.cpp
+++ b/src/rewrite_rnn.cpp
@@ -789,6 +789,11 @@ void rewrite_rnn::apply_lstm(program& prog, instruction_ref ins) const
        }
        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