How to speed up for loop in customized RNN?

Any update on this topic?