diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index 10f85ad6ce6bb..65e308a257e49 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -2866,7 +2866,7 @@ def _args_check(cls, inputs, attr, red_valids=None): @classmethod def _impl_v11(cls, inputs, attr, params): - axis = cls._args_check(inputs, attr) + axis = cls._args_check(inputs, attr)[0] return _op.scatter_elements(inputs[0], inputs[1], inputs[2], axis, "update")