fixed an error in lstm_implementation.

4702c17e · Shucai Xiao · 59f2e8e6 · 4702c17e
Commit 4702c17e authored Feb 15, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/rewrite_rnn.cpp src/rewrite_rnn.cpp +1 -1

No files found.
--- a/src/rewrite_rnn.cpp
+++ b/src/rewrite_rnn.cpp
@@ -872,7 +872,7 @@ void rewrite_rnn::apply_lstm(program& prog, instruction_ref ins) const
    while(last_output_it != ins->outputs().end())
    {
        last_output_it = std::find_if(last_output_it, ins->outputs().end(), [](auto i) {
-            return i->name() == "lstm_last_output";
+            return i->name() == "rnn_last_output";
        });

        if(last_output_it != ins->outputs().end())