This is an automated email from the ASF dual-hosted git repository.
leandron pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 13b487fa2d [TFLite][Frontend] Support quantized ELU (#15821)
13b487fa2d is described below
commit 13b487fa2d3c06a12cd9d76ec2e5a392b5eeb778
Author: Tlopex <[email protected]>
AuthorDate: Wed Oct 11 23:21:50 2023 +0800
[TFLite][Frontend] Support quantized ELU (#15821)
* add quantized elu
* add tests for quantized elu
---
python/tvm/relay/frontend/tflite.py | 2 --
tests/python/frontend/tflite/test_forward.py | 2 +-
2 files changed, 1 insertion(+), 3 deletions(-)
diff --git a/python/tvm/relay/frontend/tflite.py
b/python/tvm/relay/frontend/tflite.py
index 532318b804..cbf08da6a1 100644
--- a/python/tvm/relay/frontend/tflite.py
+++ b/python/tvm/relay/frontend/tflite.py
@@ -1275,8 +1275,6 @@ class OperatorConverter(object):
def convert_elu(self, op):
"""Convert TFLite ELU"""
- if self.is_quantized(op):
- raise tvm.error.OpNotImplemented("TFlite quantized ELU operator is
not supported yet.")
input_tensors = self.get_input_tensors(op)
assert len(input_tensors) == 1, "input tensors length should be 1"
diff --git a/tests/python/frontend/tflite/test_forward.py
b/tests/python/frontend/tflite/test_forward.py
index f601667024..d673c22adc 100644
--- a/tests/python/frontend/tflite/test_forward.py
+++ b/tests/python/frontend/tflite/test_forward.py
@@ -2436,7 +2436,7 @@ def test_all_unary_elemwise():
# from the converter that we need to provide a custom Tan operator
# implementation.
# _test_forward_unary_elemwise(_test_tan)
- _test_forward_unary_elemwise(_test_elu, quantized=False)
+ _test_forward_unary_elemwise(_test_elu)
#######################################################################
