I am not able to find the corresponding function in tensorflow 2.x.x. I also tried doing tf.compat.v1.contrib.rnn.CoupledInputForgetGateLSTMCell() but I am getting the following error - AttributeError: module 'tensorflow._api.v2.compat.v1' has no attribute 'contrib'
More generally, where are the functions in tf.contrib located in tensorflow 2?
Have you checked tfa.rnn.LayerNormLSTMCell for your case?
It is a part of TF Addons package which is compatible with TF 2.X versions
(TF Addons package contains a subset tf.contrib* functionality)