Correct PPL compute strategy in RNN