add lstm rewrite to program execution

d53a69f6 · Shucai Xiao · 2ce55c6e · d53a69f6
Commit d53a69f6 authored Feb 08, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

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

No files found.
--- a/src/rewrite_rnn.cpp
+++ b/src/rewrite_rnn.cpp
@@ -16,11 +16,14 @@ void rewrite_rnn::apply(program& prog) const
        {
            apply_vanilla_rnn(prog, ins);
        }
+        else if(ins->name() == "gru")
-        if(ins->name() == "gru")
        {
            apply_gru(prog, ins);
        }
+        else if(ins->name() == "lstm")
+        {
+            apply_lstm(prog, ins);
+        }
    }
 }