Skip to content

Commit 13b487f

Browse files
authored
[TFLite][Frontend] Support quantized ELU (#15821)
* add quantized elu * add tests for quantized elu
1 parent 9f562b0 commit 13b487f

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

python/tvm/relay/frontend/tflite.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1275,8 +1275,6 @@ def convert_neg(self, op):
12751275

12761276
def convert_elu(self, op):
12771277
"""Convert TFLite ELU"""
1278-
if self.is_quantized(op):
1279-
raise tvm.error.OpNotImplemented("TFlite quantized ELU operator is not supported yet.")
12801278
input_tensors = self.get_input_tensors(op)
12811279
assert len(input_tensors) == 1, "input tensors length should be 1"
12821280

tests/python/frontend/tflite/test_forward.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2436,7 +2436,7 @@ def test_all_unary_elemwise():
24362436
# from the converter that we need to provide a custom Tan operator
24372437
# implementation.
24382438
# _test_forward_unary_elemwise(_test_tan)
2439-
_test_forward_unary_elemwise(_test_elu, quantized=False)
2439+
_test_forward_unary_elemwise(_test_elu)
24402440

24412441

24422442
#######################################################################

0 commit comments

Comments
 (0)