Rollback more APIs for tf2-alpha

tensorflow · Apr 30, 2019 · ac69a4f · ac69a4f
1 parent a81063b
commit ac69a4f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorflow_addons/rnn/cell.py b/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,

diff --git a/tensorflow_addons/rnn/cell_test.py b/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": {}