Add support for quant. mul operator in tflite frontend (#4283)

A test for qnn_mul has to be added when the qnn elemwise tests (#4282) get merged.
apache · Nov 15, 2019 · 70a3a61 · 70a3a61
1 parent 7dca655
commit 70a3a61
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
@@ -575,8 +575,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):