From 2132e80cb2ff7bf0807701753cac0e1268625809 Mon Sep 17 00:00:00 2001 From: Joshua Chia Date: Thu, 21 Jan 2021 17:49:14 +0800 Subject: [PATCH] Made tensorflow IsNan actually work IsNan was added to tensorflow.rst in fa1b859f but this commit makes IsNan actually work --- python/tvm/relay/frontend/tensorflow.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py index d5746a38582c..2c7361a7d813 100644 --- 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"),