Skip to content

Commit 877c0b2

Browse files
authored
Pure tf.functions can't have None inputs (#2837)
1 parent f79b2fd commit 877c0b2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow_addons/image/dense_image_warp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def dense_image_warp(
247247

248248
@tf.function(experimental_implements="addons:DenseImageWarp")
249249
def dense_image_warp_annotated(
250-
image: types.TensorLike, flow: types.TensorLike, name: Optional[str] = None
250+
image: types.TensorLike, flow: types.TensorLike
251251
) -> tf.Tensor:
252252
"""Similar to dense_image_warp but annotated with experimental_implements.
253253
@@ -260,4 +260,4 @@ def dense_image_warp_annotated(
260260
However, with the annotation, this function cannot be used with backprop
261261
under `tf.GradientTape` objects.
262262
"""
263-
return dense_image_warp(image, flow, name)
263+
return dense_image_warp(image, flow)

0 commit comments

Comments
 (0)