diff --git a/tensorflow_addons/image/filters.py b/tensorflow_addons/image/filters.py index 7108119a1f..2017dd1f35 100644 --- a/tensorflow_addons/image/filters.py +++ b/tensorflow_addons/image/filters.py @@ -91,7 +91,7 @@ def func2(): for a in range(ch): img = image[:, :, a:a + 1] img = tf.reshape(img, [1, row, col, 1]) - slic = tf.image.extract_patches( + slic = tf.image.extract_image_patches( img, [1, filter_shapex, filter_shapey, 1], [1, 1, 1, 1], [1, 1, 1, 1], padding='SAME')