clean code

apache · Feb 10, 2023 · 8e302e7 · 8e302e7
1 parent 91af964
commit 8e302e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
@@ -2886,7 +2886,7 @@ def _args_check(cls, inputs, attr, red_valids=None):
     def _impl_v11(cls, inputs, attr, params):
         axis = cls._args_check(inputs, attr)
 
-        return _op.scatter(inputs[0], inputs[1], inputs[2], axis)
+        return _op.scatter_elements(inputs[0], inputs[1], inputs[2], axis, "update")
 
     @classmethod
     def _impl_v16(cls, inputs, attr, params):

diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
@@ -922,7 +922,7 @@ def verify_scatter_elements(in_shape, indices, axis=0, reduction="update"):
         0,
         "mul",
     )
-    # TODO(vvchernov): min and mul options are supported from 18 version, but CI supports 17 only
+    # TODO(vvchernov): min and max options are supported from 18 version, but CI supports 17 only
     # # Scatter elements with min reduction of duplicates
     # verify_scatter_elements(
     #     (3, 3, 3),
@@ -5468,7 +5468,6 @@ def verify_eyelike(indata, dynamic=False):
     "test_reduce_sum_negative_axes_keepdims_example",
     "test_reduce_sum_negative_axes_keepdims_random",
     "test_roialign_aligned_true",
-    "test_scatter_elements_with_duplicate_indices",
     "test_scatternd_add",
     "test_scatternd_multiply",
     "test_sequence_insert_at_back",