[LSTMCELL] Fix compile error
authorJihoon Lee <jhoon.it.lee@samsung.com>
Tue, 26 Oct 2021 01:22:03 +0000 (10:22 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Tue, 26 Oct 2021 02:59:12 +0000 (11:59 +0900)
This patch fix compile error in the main branch

**Self evaluation:**
1. Build test: [X]Passed [ ]Failed [ ]Skipped
2. Run test: [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: Jihoon Lee <jhoon.it.lee@samsung.com>
nntrainer/layers/lstmcell.cpp

index e17e196..51984ed 100644 (file)
@@ -227,7 +227,7 @@ void LSTMCellLayer::forwarding(RunLayerContext &context, bool training) {
 
   if (dropout_rate > epsilon && training) {
     Tensor &mask_ = context.getTensor(wt_idx[LSTMParams::dropout_mask]);
-    mask_ = hs.dropout_mask(dropout_rate);
+    hs.dropout_mask(dropout_rate);
     hs.multiply_i(mask_);
   }