From f7f78ae236e623a57919f9450e8b2043e681ddb3 Mon Sep 17 00:00:00 2001
From: Jeremy Johnson <jeremy.johnson@arm.com>
Date: Wed, 25 May 2022 15:26:38 +0100
Subject: Add support for uint16_t to RESCALE

Update ref-model RESCALE op to support UINT16 conversions
Add testing for RESCALE UINT16 and ERROR_IFs

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ic6e6e53de1f0b054bedb9e6ba3856e7475498aba
---
 reference_model/src/ops/template_types.h | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

(limited to 'reference_model/src/ops/template_types.h')

diff --git a/reference_model/src/ops/template_types.h b/reference_model/src/ops/template_types.h
index 0fe9a41..2bc7e04 100644
--- a/reference_model/src/ops/template_types.h
+++ b/reference_model/src/ops/template_types.h
@@ -23,7 +23,7 @@ using namespace tosa;
 
 namespace TosaReference
 {
-// Shorter aliase templates for common Eigen::Tensor types
+// Shorter alias templates for common Eigen::Tensor types
 template <typename T>
 using ETensor0 = Eigen::Tensor<T, 0>;
 template <typename T>
@@ -89,6 +89,11 @@ struct GetEigenType<DType_UINT8>
     using type = int32_t;
 };
 template <>
+struct GetEigenType<DType_UINT16>
+{
+    using type = int32_t;
+};
+template <>
 struct GetEigenType<DType_INT4>
 {
     using type = int32_t;
@@ -121,6 +126,11 @@ struct GetNumBits<DType_UINT8>
     static constexpr int32_t value = 8;
 };
 template <>
+struct GetNumBits<DType_UINT16>
+{
+    static constexpr int32_t value = 16;
+};
+template <>
 struct GetNumBits<DType_INT4>
 {
     static constexpr int32_t value = 4;
@@ -158,6 +168,11 @@ struct GetQMin<DType_UINT8>
     static constexpr int64_t value = 0L;
 };
 template <>
+struct GetQMin<DType_UINT16>
+{
+    static constexpr int64_t value = 0L;
+};
+template <>
 struct GetQMin<DType_INT4>
 {
     static constexpr int64_t value = -8L;
@@ -194,6 +209,11 @@ struct GetQMax<DType_UINT8>
     static constexpr int64_t value = 255L;
 };
 template <>
+struct GetQMax<DType_UINT16>
+{
+    static constexpr int64_t value = 65535L;
+};
+template <>
 struct GetQMax<DType_INT4>
 {
     static constexpr int64_t value = 7L;
-- 
cgit v1.2.1