Add support for quant. mul operator in tflite frontend

A test for qnn_mul has to be added when the qnn elemwise tests (apache#4282) get merged.
inadob · Nov 8, 2019 · 83eb895 · 83eb895
1 parent 281f643
commit 83eb895
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py
@@ -574,8 +574,7 @@ def convert_mul(self, op):
         """Convert TFLite MUL"""
         # Check if the input tensor is quantized, call QNN op
         if self.is_quantized(op):
-            raise tvm.error.OpNotImplemented(
-                'TFlite quantized mul operator is not supported yet.')
+            return self._convert_elemwise(_qnn.op.mul, op)
         return self._convert_elemwise(_op.multiply, op)
 
     def convert_div(self, op):