Add support for quant. mul operator in tflite frontend (#4283)
authorIna Dobreva <55383260+inadob@users.noreply.github.com>
Fri, 15 Nov 2019 03:41:36 +0000 (03:41 +0000)
committerTianqi Chen <tqchen@users.noreply.github.com>
Fri, 15 Nov 2019 03:41:36 +0000 (19:41 -0800)
A test for qnn_mul has to be added when the qnn elemwise tests (#4282) get merged.

python/tvm/relay/frontend/tflite.py

index 8966aa6..30209ff 100644 (file)
@@ -575,8 +575,7 @@ class OperatorConverter(object):
         """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):