Skip to content

Commit 300e5e0

Browse files
inadobWei Chen
authored andcommitted
add parser support for GREATER tflite operator (apache#3963)
add test for GREATER
1 parent 58ed618 commit 300e5e0

2 files changed

Lines changed: 12 additions & 0 deletions

File tree

python/tvm/relay/frontend/tflite.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@ def __init__(self, model, subgraph, exp_tab):
7474
'POW': self.convert_pow,
7575
'MAXIMUM': self.convert_maximum,
7676
'MINIMUM': self.convert_minimum,
77+
'GREATER': self.convert_greater,
7778
'REDUCE_MIN': self._convert_reduce_min,
7879
'REDUCE_MAX': self._convert_reduce_max,
7980
'MEAN': self._convert_reduce_mean,
@@ -456,6 +457,9 @@ def convert_maximum(self, op):
456457
def convert_minimum(self, op):
457458
return self._convert_elemwise(_op.minimum, op)
458459

460+
def convert_greater(self, op):
461+
return self._convert_elemwise(_op.greater, op)
462+
459463
def _convert_reduce(self, relay_op, op):
460464
"""Generic method to Convert TFLite MEAN operators"""
461465
try:

tests/python/frontend/tflite/test_forward.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -597,6 +597,13 @@ def _test_maximum(data):
597597
def _test_minimum(data):
598598
""" One iteration of minimum """
599599
return _test_elemwise(math_ops.minimum, data)
600+
#######################################################################
601+
# Greater
602+
# -------
603+
604+
def _test_greater(data):
605+
""" One iteration of greater """
606+
return _test_elemwise(math_ops.greater, data)
600607

601608
def _test_forward_elemwise(testop):
602609
""" Elewise"""
@@ -623,6 +630,7 @@ def test_all_elemwise():
623630
_test_forward_elemwise(_test_pow)
624631
_test_forward_elemwise(_test_maximum)
625632
_test_forward_elemwise(_test_minimum)
633+
_test_forward_elemwise(_test_greater)
626634

627635
#######################################################################
628636
# Reduce

0 commit comments

Comments
 (0)