Skip to content

Commit 251bd94

Browse files
author
Valery Chernov
committed
clean code
1 parent 0889a50 commit 251bd94

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

python/tvm/relay/frontend/onnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2886,7 +2886,7 @@ def _args_check(cls, inputs, attr, red_valids=None):
28862886
def _impl_v11(cls, inputs, attr, params):
28872887
axis = cls._args_check(inputs, attr)
28882888

2889-
return _op.scatter(inputs[0], inputs[1], inputs[2], axis)
2889+
return _op.scatter_elements(inputs[0], inputs[1], inputs[2], axis, "update")
28902890

28912891
@classmethod
28922892
def _impl_v16(cls, inputs, attr, params):

tests/python/frontend/onnx/test_forward.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -922,7 +922,7 @@ def verify_scatter_elements(in_shape, indices, axis=0, reduction="update"):
922922
0,
923923
"mul",
924924
)
925-
# TODO(vvchernov): min and mul options are supported from 18 version, but CI supports 17 only
925+
# TODO(vvchernov): min and max options are supported from 18 version, but CI supports 17 only
926926
# # Scatter elements with min reduction of duplicates
927927
# verify_scatter_elements(
928928
# (3, 3, 3),
@@ -5468,7 +5468,6 @@ def verify_eyelike(indata, dynamic=False):
54685468
"test_reduce_sum_negative_axes_keepdims_example",
54695469
"test_reduce_sum_negative_axes_keepdims_random",
54705470
"test_roialign_aligned_true",
5471-
"test_scatter_elements_with_duplicate_indices",
54725471
"test_scatternd_add",
54735472
"test_scatternd_multiply",
54745473
"test_sequence_insert_at_back",

0 commit comments

Comments
 (0)