fix a bug in implementation

59f2e8e6 · Shucai Xiao · b45bfed7 · 59f2e8e6
Commit 59f2e8e6 authored Feb 14, 2019 by Shucai Xiao
Show 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
@@ -980,7 +980,7 @@ std::vector<instruction_ref> rewrite_rnn::lstm_cell(bool is_forward,
        bo_brcst = prog.insert_instruction(ins, op::broadcast{1, ic_shape}, bo);
        auto bxf = prog.insert_instruction(ins, op::slice{{0}, {2 * hs}, {3 * hs}}, sbias);
-        auto bhf = prog.insert_instruction(ins, op::slice{{0}, {5 * hs}, {6 * hs}}, sbias);
+        auto bhf = prog.insert_instruction(ins, op::slice{{0}, {6 * hs}, {7 * hs}}, sbias);
        auto bf  = prog.insert_instruction(ins, op::add{}, bxf, bhf);
        bf_brcst = prog.insert_instruction(ins, op::broadcast{1, ic_shape}, bf);