Rollback more APIs for tf2-alpha

Author: seanpmorgan

tensorflow_addons/rnn/cell.py

@@ -363,7 +363,7 @@ def get_config(self):
         return dict(list(base_config.items()) + list(config.items()))
 
     def _create_norm_layer(self, name):
-        return keras.layers.LayerNormalization(
+        return keras.layers.experimental.LayerNormalization(
             beta_initializer=self.norm_beta_initializer,
             gamma_initializer=self.norm_gamma_initializer,
             epsilon=self.norm_epsilon,

tensorflow_addons/rnn/cell_test.py

@@ -274,7 +274,7 @@ def test_config(self):
             "bias_constraint": None,
             "dropout": 0.,
             "recurrent_dropout": 0.,
-            "implementation": 2,
+            "implementation": 1,
             "norm_gamma_initializer": {
                 "class_name": "Ones",
                 "config": {}