diff --git a/imperative/python/tools/gen_ops.py b/imperative/python/tools/gen_ops.py index dde93e98..299b3c19 100755 --- a/imperative/python/tools/gen_ops.py +++ b/imperative/python/tools/gen_ops.py @@ -257,7 +257,7 @@ class Context: def decl_raw_opr(self, name, *, inputs, inputs_cvt=[], body=None, - desc=None, local_defs=[], have_config=True): + desc=None, local_defs=[], have_config=True, params=None, has_out_dtype=False): self.skipped.append(name) def get_str(self):