commit 0ca9520c35cd10cff494ba9cb4c4725694b561b1
author Lev Proleev <levp@google.com> Wed Oct 23 11:40:41 2019 +0100
committer Slava Shklyaev <slavash@google.com> Tue Nov 26 17:44:01 2019 +0000
tree 42d19338020a1c38a4917e8dd0515102dfba200c
parent ef642207d58e466112cae470742607be6af3d3a5

Add QUANT8_ASYMM_SIGNED support to SELECT op

Bug: 143935354
Test: NNTest_static and VTS_1_3  with --gtest_filter="*select*"
Change-Id: Ifc8ae06d412d4b50421a4e2dce39b223723a137a
Merged-In: Ifc8ae06d412d4b50421a4e2dce39b223723a137a
(cherry picked from commit 29c9adde6ec7ea99c98b6d796c5a79c19dea8701)

common/OperationsUtils.cpp

diff --git a/common/OperationsUtils.cpp b/common/OperationsUtils.cpp
index 5c50e3d..5c251a2 100644
--- a/common/OperationsUtils.cpp
+++ b/common/OperationsUtils.cpp
@@ -17,12 +17,16 @@
 #define LOG_TAG "OperationsUtils"
 
 #include "OperationsUtils.h"
+
+#include <algorithm>
+#include <cmath>
+#include <limits>
+#include <sstream>
+#include <vector>
+
 #include "Operations.h"
 #include "Utils.h"
 
-#include <cmath>
-#include <sstream>
-
 namespace android {
 namespace nn {
 
@@ -355,6 +359,14 @@
     return static_cast<uint8_t>(std::round(doubleRet));
 }
 
+int8_t requantize(int8_t value, const Shape& oldShape, const Shape& newShape) {
+    double doubleValue = (value - oldShape.offset) * oldShape.scale;
+    double doubleRet = doubleValue / newShape.scale + newShape.offset;
+    if (doubleRet < -128) return -128;
+    if (doubleRet > 127) return 127;
+    return static_cast<int8_t>(std::round(doubleRet));
+}
+
 bool floorPrepare(const Shape& input, Shape* output) {
     return SetShape(input, output);
 }