Change newly added atan

apache · Mar 20, 2020 · f987edd · f987edd
1 parent 48dd312
commit f987edd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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