Merge pull request #817 from Kaggle/add-tfa-custom-op-test

Add a test exercising TFA custom op.

Author: rosbo

tests/test_tensorflow_addons.py

@@ -11,4 +11,12 @@ def test_tfa_image(self):
         img = tf.image.convert_image_dtype(img, tf.float32)
         mean = tfa.image.mean_filter2d(img, filter_shape=1)
 
-        self.assertEqual(1, len(mean))
+        self.assertEqual(1, len(mean))
+    
+    # This test exercises TFA Custom Op. See: b/145555176
+    def test_gelu(self):
+        x = tf.constant([[0.5, 1.2, -0.3]])
+        layer = tfa.layers.GELU()
+        result = layer(x)
+
+        self.assertEqual((1, 3), result.shape)