[layer] Bug fix for LSTMCell
authorParichay Kapoor <pk.kapoor@samsung.com>
Mon, 25 Oct 2021 01:29:50 +0000 (10:29 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Tue, 26 Oct 2021 00:11:31 +0000 (09:11 +0900)
commit59223093b78c1e0c3bf8e564e9bc7b05a6cffb1a
tree97ecb4565358c9a88ea7d68b3f3051fdb4d57c3e
parent371953f4b46fb4d7c3ad61a933c41b8a8c15b32e
[layer] Bug fix for LSTMCell

Enable recurrent unrolling tests for LSTMCell with a single and stacked
LSTM cell. Add corresponding bug fixes.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>
jni/Android.mk
nntrainer/compiler/recurrent_realizer.cpp
nntrainer/layers/lstmcell.cpp
nntrainer/tensor/tensor.cpp
test/unittest/models/unittest_models_recurrent.cpp