Merge pull request #817 from Kaggle/add-tfa-custom-op-test · spenfraz/docker-python@2b66611 · GitHub

spenfraz / docker-python Public

forked from Kaggle/docker-python

• Notifications You must be signed in to change notification settings
• Fork 0
• Star 0

• Code
• Pull requests
• Actions
• Projects
• Security and quality
• Insights

Additional navigation options

Commit 2b66611

Browse filesBrowse the repository at this point in the historyBrowse files

rosbo

authored

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

Add a test exercising TFA custom op.

2 parents 57fb90f + 09a76b8 commit 2b66611
Copy full SHA for 2b66611

1 file changed

+9-1Lines changed: 9 additions & 1 deletion

File tree

Expand file treeCollapse file tree

Open diff view settings

Filter options

• tests
  • test_tensorflow_addons.py

Expand file treeCollapse file tree

Open diff view settings

Collapse file

‎tests/test_tensorflow_addons.py‎

Copy file name to clipboardExpand all lines: tests/test_tensorflow_addons.py

+9-1Lines changed: 9 additions & 1 deletion

Original file line number	Diff line number	Diff line change
@@ -11,4 +11,12 @@ def test_tfa_image(self):
11	11	img = tf.image.convert_image_dtype(img, tf.float32)
12	12	mean = tfa.image.mean_filter2d(img, filter_shape=1)
13	13
14		- self.assertEqual(1, len(mean))
	14	+ self.assertEqual(1, len(mean))
	15	+
	16	+ # This test exercises TFA Custom Op. See: b/145555176
	17	+ def test_gelu(self):
	18	+ x = tf.constant([[0.5, 1.2, -0.3]])
	19	+ layer = tfa.layers.GELU()
	20	+ result = layer(x)
	21	+
	22	+ self.assertEqual((1, 3), result.shape)

0 commit comments

Comments

0 (0)