[ONNX] Fix bug in scatter_elements (apache#15017)

fix bug in scatter_elements Co-authored-by: Valery Chernov <valery.chernov@deelvin.com>
junrushao · Jun 22, 2023 · d4bfdfd · d4bfdfd
1 parent 06875d3
commit d4bfdfd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")