This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new efb09a5953 [ONNX] Fix bug in scatter_elements (#15017)
efb09a5953 is described below
commit efb09a595337203e43ba71a356705995d7e9fde7
Author: Valery Chernov <[email protected]>
AuthorDate: Sun Jun 4 15:44:50 2023 +0400
[ONNX] Fix bug in scatter_elements (#15017)
fix bug in scatter_elements
Co-authored-by: Valery Chernov <[email protected]>
---
python/tvm/relay/frontend/onnx.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/relay/frontend/onnx.py
b/python/tvm/relay/frontend/onnx.py
index 10f85ad6ce..65e308a257 100644
--- a/python/tvm/relay/frontend/onnx.py
+++ b/python/tvm/relay/frontend/onnx.py
@@ -2866,7 +2866,7 @@ class ScatterElements(OnnxOpConverter):
@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")
