From 83eb89593a20a59e8467c577e79ec6dcb50d0c2d Mon Sep 17 00:00:00 2001 From: Ina_Dobreva Date: Fri, 8 Nov 2019 15:08:45 +0000 Subject: [PATCH] Add support for quant. mul operator in tflite frontend A test for qnn_mul has to be added when the qnn elemwise tests (#4282) get merged. --- python/tvm/relay/frontend/tflite.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py index b042af9fbe65..cd17efb0daaa 100644 --- 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):