Skip to content

Commit

Permalink
[Frontend][ONNX] Add onnx support for LessOrEqual, GreaterOrEqual
Browse files Browse the repository at this point in the history
  • Loading branch information
SamKG committed Aug 3, 2021
1 parent 4b9d43e commit 17f93f8
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 0 deletions.
18 changes: 18 additions & 0 deletions python/tvm/relay/frontend/onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -1620,6 +1620,22 @@ def _impl_v7(cls, inputs, attr, params):
return _op.less(inputs[0], inputs[1])


class LessOrEqual(OnnxOpConverter):
"""Operator logical less or equal than."""

@classmethod
def _impl_v12(cls, inputs, attr, params):
return _op.less_equal(inputs[0], inputs[1])


class GreaterOrEqual(OnnxOpConverter):
"""Operator logical greater or equal than."""

@classmethod
def _impl_v12(cls, inputs, attr, params):
return _op.greater_equal(inputs[0], inputs[1])


class LRN(OnnxOpConverter):
"""Operator converter for Local Response Normalization."""

Expand Down Expand Up @@ -3477,6 +3493,8 @@ def _get_convert_map(opset):
"Exp": Renamer("exp"),
"Greater": Greater.get_converter(opset),
"Less": Less.get_converter(opset),
"LessOrEqual": LessOrEqual.get_converter(opset),
"GreaterOrEqual": GreaterOrEqual.get_converter(opset),
"Log": Renamer("log"),
"Acos": Renamer("acos"),
"Acosh": Renamer("acosh"),
Expand Down
4 changes: 4 additions & 0 deletions tests/python/frontend/onnx/test_forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -1927,8 +1927,12 @@ def verify_binary_ops(op, x, y, out_type="float32"):
verify_binary_ops("Sum", x, z)
verify_binary_ops("Greater", x, y, "bool")
verify_binary_ops("Greater", x, z, "bool")
verify_binary_ops("GreaterOrEqual", x, y, "bool")
verify_binary_ops("GreaterOrEqual", x, z, "bool")
verify_binary_ops("Less", x, y, "bool")
verify_binary_ops("Less", x, z, "bool")
verify_binary_ops("LessOrEqual", x, y, "bool")
verify_binary_ops("LessOrEqual", x, z, "bool")
verify_binary_ops("Equal", x, y, "bool")
verify_binary_ops("Equal", x, z, "bool")

Expand Down

0 comments on commit 17f93f8

Please sign in to comment.