Test with tf.map_fn

tensorflow · May 14, 2020 · 078b019 · 078b019
1 parent fe84ea9
commit 078b019
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow/python/autograph/impl/api_test.py b/tensorflow/python/autograph/impl/api_test.py
@@ -294,7 +294,8 @@ def test_converted_call_functools_partial_decorated(self):
 
     @api.convert(recursive=DEFAULT_RECURSIVE)
     def test_call_partial():
-      return functools.partial(test_fn, constant_op.constant(-1), y=-2, z=-3)
+      fn = functools.partial(test_fn)
+      return tf.map_fn(fn, constant_op.constant(-1), y=-2, z=-3)
 
     def test_fn(x, y, z):
       if x < 0: