Fix floating value quantization for RELU6 and RELU1

apache · Jun 16, 2020 · 9345405 · 9345405
1 parent 7f0599c
commit 9345405
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py
@@ -702,8 +702,9 @@ def convert_relu6(self, op):
 
         if input_tensor.qnn_params:
             # Quantize a float value to an quantized integer value
-            quantize = lambda x: float(int(round(x / input_tensor.qnn_params['scale'])) + \
-                                       input_tensor.qnn_params['zero_point'])
+            scale_val = get_scalar_from_constant(input_tensor.qnn_params['scale'])
+            zero_point_val = get_scalar_from_constant(input_tensor.qnn_params['zero_point'])
+            quantize = lambda x: float(int(round(x / scale_val)) + zero_point_val)
 
             # Get min/max of the input dtype. This will be used to ensure that
             # clip a_min/a_max are not beyond the dtype range.
@@ -772,8 +773,9 @@ def convert_relu_n1_to_1(self, op):
 
         if input_tensor.qnn_params:
             # Quantize a float value to an quantized integer value
-            quantize = lambda x: float(int(round(x / input_tensor.qnn_params['scale'])) + \
-                                       input_tensor.qnn_params['zero_point'])
+            scale_val = get_scalar_from_constant(input_tensor.qnn_params['scale'])
+            zero_point_val = get_scalar_from_constant(input_tensor.qnn_params['zero_point'])
+            quantize = lambda x: float(int(round(x / scale_val)) + zero_point_val)
 
             # Get min/max of the input dtype. This will be used to ensure that
             # clip a_min/a_max are not beyond the dtype range.