[layer] LSTM cell batchresize bug fix
authorParichay Kapoor <pk.kapoor@samsung.com>
Tue, 26 Oct 2021 07:06:11 +0000 (16:06 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Thu, 28 Oct 2021 01:28:03 +0000 (10:28 +0900)
commit2ab73cd8cc523eeaed773feabbb5b50cd69025ba
treecd3d7775b83a9d450d7325a57587370aa67ba475
parent630526f943f03604e736e9e6bae98706c5591fdd
[layer] LSTM cell batchresize bug fix

As LSTM cell uses the batch values not in the batch dimension but in the
inner dimension, it creates bug with the existing layer context interface
as it only allows updating the batch along the batch dimension.
This patch provides a quick fix to combine the time and batch values in
the batch dimension.

Once recurrent realizer allows multiple inputs, the time factor will be
removed and the reshapes can be removed.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>
nntrainer/layers/lstmcell.cpp