vender lstm now not support reverse mode

PaddlePaddle · Aug 1, 2018 · ff84ec5 · ff84ec5
1 parent 8e883be
commit ff84ec5
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/run.sh b/run.sh
diff --git a/test/saber/cuda/test_saber_func_lstm.cpp b/test/saber/cuda/test_saber_func_lstm.cpp
@@ -27,7 +27,7 @@ void test_saber_lstm(int sequence_size = 2, int batch_size = 1, int word_size =
 
     //std::vector<int> offsets = {0, 20,40, 65, 82, 101};
     std::vector<int> offsets = {0, 3};
-    bool is_reverse = true;
+    bool is_reverse = false;
     batch_size = offsets.size() - 1;
     Shape input_shape(offsets[offsets.size() - 1], word_size, 1, 1);
     Shape output_shape(offsets[offsets.size() - 1], hidden_size, 1, 1);