LSTM: pulled out the linear parts in scan op