aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/BaseIterator.hpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2019-04-09 14:20:12 +0100
committerSadik Armagan <sadik.armagan@arm.com>2019-04-09 14:24:05 +0100
commit2999a02f0c6a6f290ce45f28c998a1c000d48f67 (patch)
treef9d13cec08ab8c6c47e68df512cddc613552a7d2 /src/backends/reference/workloads/BaseIterator.hpp
parent998517647d699d602e36f06b40d3f1d1ddaae7be (diff)
downloadarmnn-2999a02f0c6a6f290ce45f28c998a1c000d48f67.tar.gz
IVGCVSW-2862 Extend the Elementwise Workload to support QSymm16 Data Type
IVGCVSW-2863 Unit test per Elementwise operator with QSymm16 Data Type * Added QSymm16 support for Elementwise Operators * Added QSymm16 unit tests for Elementwise Operators Change-Id: I4e4e2938f9ed2cbbb1f05fb0f7dc476768550277 Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Diffstat (limited to 'src/backends/reference/workloads/BaseIterator.hpp')
-rw-r--r--src/backends/reference/workloads/BaseIterator.hpp32
1 files changed, 32 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/BaseIterator.hpp b/src/backends/reference/workloads/BaseIterator.hpp
index cfa8ce7e91..95c75a576a 100644
--- a/src/backends/reference/workloads/BaseIterator.hpp
+++ b/src/backends/reference/workloads/BaseIterator.hpp
@@ -112,6 +112,22 @@ public:
}
};
+class QSymm16Decoder : public TypedIterator<const int16_t, Decoder>
+{
+public:
+ QSymm16Decoder(const int16_t* data, const float scale, const int32_t offset)
+ : TypedIterator(data), m_Scale(scale), m_Offset(offset) {}
+
+ float Get() const override
+ {
+ return armnn::Dequantize(*m_Iterator, m_Scale, m_Offset);
+ }
+
+private:
+ const float m_Scale;
+ const int32_t m_Offset;
+};
+
class FloatEncoder : public TypedIterator<float, Encoder>
{
public:
@@ -152,4 +168,20 @@ public:
}
};
+class QSymm16Encoder : public TypedIterator<int16_t, Encoder>
+{
+public:
+ QSymm16Encoder(int16_t* data, const float scale, const int32_t offset)
+ : TypedIterator(data), m_Scale(scale), m_Offset(offset) {}
+
+ void Set(const float& right) override
+ {
+ *m_Iterator = armnn::Quantize<int16_t>(right, m_Scale, m_Offset);
+ }
+
+private:
+ const float m_Scale;
+ const int32_t m_Offset;
+};
+
} //namespace armnn \ No newline at end of file