Skip to content

Commit 548675f

Browse files
AndrewZhaoLuoAndrew Zhao Luo
andauthored
[EZ] [ONNX] Remove unnecessary converters for greater and lesser (#8967)
* simplify by removing unneeded conversions * remove uneeded * remove test Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>
1 parent 1bebd0a commit 548675f

File tree

1 file changed

+2
-18
lines changed

1 file changed

+2
-18
lines changed

python/tvm/relay/frontend/onnx.py

Lines changed: 2 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1611,22 +1611,6 @@ def _impl_v9(cls, inputs, attr, params):
16111611
return _op.scatter_nd(zeros, _op.stack([indices, indices + k], axis=0), ones, "update")
16121612

16131613

1614-
class Greater(OnnxOpConverter):
1615-
"""Operator logical greater."""
1616-
1617-
@classmethod
1618-
def _impl_v7(cls, inputs, attr, params):
1619-
return _op.greater(inputs[0], inputs[1])
1620-
1621-
1622-
class Less(OnnxOpConverter):
1623-
"""Operator logical less than."""
1624-
1625-
@classmethod
1626-
def _impl_v7(cls, inputs, attr, params):
1627-
return _op.less(inputs[0], inputs[1])
1628-
1629-
16301614
class LRN(OnnxOpConverter):
16311615
"""Operator converter for Local Response Normalization."""
16321616

@@ -3655,8 +3639,8 @@ def _get_convert_map(opset):
36553639
"Selu": Selu.get_converter(opset),
36563640
"Elu": Elu.get_converter(opset),
36573641
"Exp": Renamer("exp"),
3658-
"Greater": Greater.get_converter(opset),
3659-
"Less": Less.get_converter(opset),
3642+
"Greater": Renamer("greater"),
3643+
"Less": Renamer("less"),
36603644
"Log": Renamer("log"),
36613645
"Acos": Renamer("acos"),
36623646
"Acosh": Renamer("acosh"),

0 commit comments

Comments
 (0)