How to use SRU model in a GRU-model?

[dawn0815]

Hello! Thank you for your work in optimizing the RNN model.Now,I just want to use your SRU model to alternative the GRU model in my project,but I have a problem:just like you said,I can use SRU to exchange LSTM,but I can't use SRUCell to alternative GRUCell.What can I do? Thank you!