Skip to content

Commit 169e0d6

Browse files
committed
add parser support for GREATER tflite operator
add test for GREATER
1 parent b0ddcff commit 169e0d6

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
@@ -73,6 +73,7 @@ def __init__(self, model, subgraph, exp_tab):
7373
'POW': self.convert_pow,
7474
'MAXIMUM': self.convert_maximum,
7575
'MINIMUM': self.convert_minimum,
76+
'GREATER': self.convert_greater,
7677
'REDUCE_MIN': self._convert_reduce_min,
7778
'REDUCE_MAX': self._convert_reduce_max,
7879
'MEAN': self._convert_reduce_mean,
@@ -435,6 +436,9 @@ def convert_maximum(self, op):
435436
def convert_minimum(self, op):
436437
return self._convert_elemwise(_op.minimum, op)
437438

439+
def convert_greater(self, op):
440+
return self._convert_elemwise(_op.greater, op)
441+
438442
def _convert_reduce(self, relay_op, op):
439443
"""Generic method to Convert TFLite MEAN operators"""
440444
try:

tests/python/frontend/tflite/test_forward.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -520,6 +520,13 @@ def _test_maximum(data):
520520
def _test_minimum(data):
521521
""" One iteration of minimum """
522522
return _test_elemwise(math_ops.minimum, data)
523+
#######################################################################
524+
# Greater
525+
# -------
526+
527+
def _test_greater(data):
528+
""" One iteration of greater """
529+
return _test_elemwise(math_ops.greater, data)
523530

524531
def _test_forward_elemwise(testop):
525532
""" Elewise"""
@@ -546,6 +553,7 @@ def test_all_elemwise():
546553
_test_forward_elemwise(_test_pow)
547554
_test_forward_elemwise(_test_maximum)
548555
_test_forward_elemwise(_test_minimum)
556+
_test_forward_elemwise(_test_greater)
549557

550558
#######################################################################
551559
# Reduce

0 commit comments

Comments
 (0)