From f987eddbf0bb0210f53710bb35dc5b24b2d408ce Mon Sep 17 00:00:00 2001 From: Yao Wang Date: Fri, 20 Mar 2020 23:21:46 +0000 Subject: [PATCH] Change newly added atan --- python/tvm/relay/frontend/tensorflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py index d68f4bd7e731e..3316a574604ea 100644 --- a/python/tvm/relay/frontend/tensorflow.py +++ b/python/tvm/relay/frontend/tensorflow.py @@ -1604,8 +1604,8 @@ def _impl(inputs, attr, params, mod): return _impl def _atan2(): - def _impl(inputs, attr, params): - divide = _elemwise("divide")(inputs, attr, params) + def _impl(inputs, attr, params, mod): + divide = _elemwise("divide")(inputs, attr, params, mod) return get_relay_op("atan")(divide) return _impl