From 35d01d53d1dc84c819e38a2f6c365590aae7de87 Mon Sep 17 00:00:00 2001 From: Valery Chernov Date: Fri, 10 Feb 2023 08:16:09 +0300 Subject: [PATCH] clean code --- python/tvm/relay/frontend/onnx.py | 2 +- tests/python/frontend/onnx/test_forward.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index b5c3ce2fb85b1..ffbfbba000f84 100644 --- 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 index 0a215428b1d4f..880735dfa3733 100644 --- 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",