Internal change

PiperOrigin-RevId: 485670162

Author: mattdangerw

official/modeling/activations/gelu_test.py

@@ -16,12 +16,10 @@
 
 import tensorflow as tf
 
-from tensorflow.python.keras import keras_parameterized  # pylint: disable=g-direct-tensorflow-import
 from official.modeling import activations
 
 
-@keras_parameterized.run_all_keras_modes
-class GeluTest(keras_parameterized.TestCase):
+class GeluTest(tf.test.TestCase):
 
   def test_gelu(self):
     expected_data = [[0.14967535, 0., -0.10032465],

official/modeling/activations/mish_test.py

@@ -16,12 +16,10 @@
 
 import tensorflow as tf
 
-from tensorflow.python.keras import keras_parameterized  # pylint: disable=g-direct-tensorflow-import
 from official.modeling import activations
 
 
-@keras_parameterized.run_all_keras_modes
-class MishTest(keras_parameterized.TestCase):
+class MishTest(tf.test.TestCase):
 
   def test_mish(self):
     x = tf.constant([1.0, 0.0])

official/modeling/activations/relu_test.py

@@ -16,13 +16,10 @@
 
 import tensorflow as tf
 
-from tensorflow.python.keras import \
-  keras_parameterized  # pylint: disable=g-direct-tensorflow-import
 from official.modeling import activations
 
 
-@keras_parameterized.run_all_keras_modes
-class CustomizedReluTest(keras_parameterized.TestCase):
+class CustomizedReluTest(tf.test.TestCase):
 
   def test_relu6(self):
     features = [[.25, 0, -.25], [-1, -2, 3]]

official/modeling/activations/sigmoid_test.py

@@ -17,13 +17,10 @@
 import numpy as np
 import tensorflow as tf
 
-from tensorflow.python.keras import \
-  keras_parameterized  # pylint: disable=g-direct-tensorflow-import
 from official.modeling import activations
 
 
-@keras_parameterized.run_all_keras_modes
-class CustomizedSigmoidTest(keras_parameterized.TestCase):
+class CustomizedSigmoidTest(tf.test.TestCase):
 
   def _hard_sigmoid_nn(self, x):
     x = np.float32(x)

official/modeling/activations/swish_test.py

@@ -16,12 +16,10 @@
 import numpy as np
 import tensorflow as tf
 
-from tensorflow.python.keras import keras_parameterized  # pylint: disable=g-direct-tensorflow-import
 from official.modeling import activations
 
 
-@keras_parameterized.run_all_keras_modes
-class CustomizedSwishTest(keras_parameterized.TestCase):
+class CustomizedSwishTest(tf.test.TestCase):
 
   def _hard_swish_np(self, x):
     x = np.float32(x)