Made tensorflow IsNan actually work

IsNan was added to tensorflow.rst in fa1b859 but this commit makes IsNan actually work
apache · Jan 21, 2021 · 2132e80 · 2132e80
1 parent 20e03bc
commit 2132e80
Showing 1 changed file with 1 addition 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"),