Made tensorflow IsNan actually work (apache#7320)

* Made tensorflow IsNan actually work IsNan was added to tensorflow.rst in fa1b859 but this commit makes IsNan actually work * Added test case for tensorflow.is_nan
electriclilies · Feb 18, 2021 · aaa87bf · aaa87bf
1 parent 433e4b6
commit aaa87bf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py
@@ -2357,6 +2357,7 @@ def _impl(inputs, attr, params, mod):
     "Identity": _identity(),
     "IsFinite": AttrCvt("isfinite"),
     "IsInf": AttrCvt("isinf"),
+    "IsNan": AttrCvt("isnan"),
     "LeakyRelu": AttrCvt("leaky_relu"),
     "LeftShift": AttrCvt("left_shift"),
     "Less": _broadcast("less"),

diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py
@@ -4178,6 +4178,10 @@ def test_forward_isfinite():
     _verify_infiniteness_ops(tf.is_finite, "isfinite")
 
 
+def test_forward_isnan():
+    _verify_infiniteness_ops(tf.is_nan, "isnan")
+
+
 def _test_spop_placeholder_without_shape_info():
     with tf.Graph().as_default():