projects
/
platform
/
core
/
ml
/
nntrainer.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
5922309
)
[LSTMCELL] Fix compile error
author
Jihoon Lee
<jhoon.it.lee@samsung.com>
Tue, 26 Oct 2021 01:22:03 +0000
(10:22 +0900)
committer
Jijoong 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
patch
|
blob
|
history
diff --git
a/nntrainer/layers/lstmcell.cpp
b/nntrainer/layers/lstmcell.cpp
index
e17e196
..
51984ed
100644
(file)
--- a/
nntrainer/layers/lstmcell.cpp
+++ b/
nntrainer/layers/lstmcell.cpp
@@
-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_);
}