From 2a82f0e2680eef57ee8a889af0bbca5a6fde1759 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AF=85?= <liyin@xiaomi.com>
Date: Mon, 5 Nov 2018 19:11:45 +0800
Subject: [PATCH] Force finding range if specified

---
 mace/ops/quantize.cc | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/mace/ops/quantize.cc b/mace/ops/quantize.cc
index 3b2ea301..3bed43b6 100644
--- a/mace/ops/quantize.cc
+++ b/mace/ops/quantize.cc
@@ -33,7 +33,10 @@ class QuantizeOp<DeviceType::CPU, uint8_t> : public Operation {
   explicit QuantizeOp(OpConstructContext *context)
       : Operation(context),
         non_zero_(
-            static_cast<bool>(Operation::GetOptionalArg<int>("non_zero", 0))) {}
+            static_cast<bool>(Operation::GetOptionalArg<int>("non_zero", 0))),
+        find_range_every_time_(static_cast<bool>(Operation::GetOptionalArg<int>(
+            "find_range_every_time",
+            0))) {}
 
   MaceStatus Run(OpContext *context) override {
     MACE_UNUSED(context);
@@ -44,7 +47,7 @@ class QuantizeOp<DeviceType::CPU, uint8_t> : public Operation {
     Tensor::MappingGuard output_guard(output);
     const float *input_data = input->data<float>();
     uint8_t *output_data = output->mutable_data<uint8_t>();
-    if (output->scale() > 0.f) {
+    if (!find_range_every_time_ && output->scale() > 0.f) {
       QuantizeWithScaleAndZeropoint(input_data,
                                     input->size(),
                                     output->scale(),
@@ -67,6 +70,7 @@ class QuantizeOp<DeviceType::CPU, uint8_t> : public Operation {
 
  private:
   bool non_zero_;
+  bool find_range_every_time_;
 };
 
 template <DeviceType D, class T>
-- 
GitLab