Add support for int16_t inputs in eager interface

Implement support of int16_t inputs through int32_t underlying storage
buffers. Values are upcasted/downcasted as needed through explicit copy.

Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I3752d98531c859002539bf1cb65413ceeff05e95

1 files changed, 36 insertions, 0 deletions

diff --git a/reference_model/src/tensor.h b/reference_model/src/tensor.h
index 203cfec..5bcd1b2 100644
--- a/reference_model/src/tensor.h
+++ b/reference_model/src/tensor.h
@@ -241,11 +241,13 @@ public:
 
     virtual int setTensorValueDouble(const size_t bufLen, const double* vals) = 0;
     virtual int setTensorValueFloat(const size_t bufLen, const float* vals)   = 0;
+    virtual int setTensorValueInt16(const size_t bufLen, const int16_t* vals) = 0;
     virtual int setTensorValueInt32(const size_t bufLen, const int32_t* vals) = 0;
     virtual int setTensorValueInt64(const size_t bufLen, const int64_t* vals) = 0;
     virtual int setTensorValueBool(const size_t bufLen, const bool* vals)     = 0;
     virtual int getTensorValueDouble(const size_t bufLen, double* fbuf) const = 0;
     virtual int getTensorValueFloat(const size_t bufLen, float* fbuf) const   = 0;
+    virtual int getTensorValueInt16(const size_t bufLen, int16_t* ibuf) const = 0;
     virtual int getTensorValueInt32(const size_t bufLen, int32_t* ibuf) const = 0;
     virtual int getTensorValueInt64(const size_t bufLen, int64_t* ibuf) const = 0;
     virtual int getTensorValueBool(const size_t bufLen, bool* ibuf) const     = 0;
@@ -257,6 +259,7 @@ public:
     virtual int readfromVector(const ArrayProxy<double> vals);
     virtual int readfromVector(const ArrayProxy<float> vals);
     virtual int readfromVector(const ArrayProxy<half_float::half> vals);
+    virtual int readfromVector(const ArrayProxy<int16_t> vals);
     virtual int readfromVector(const ArrayProxy<int32_t> vals);
     virtual int readfromVector(const ArrayProxy<int64_t> vals);
     virtual int readfromVector(const ArrayProxy<unsigned char> vals);
@@ -264,6 +267,7 @@ public:
     virtual int writeToVector(ArrayProxy<double> vals);
     virtual int writeToVector(ArrayProxy<float> vals);
     virtual int writeToVector(ArrayProxy<half_float::half> vals);
+    virtual int writeToVector(ArrayProxy<int16_t> vals);
     virtual int writeToVector(ArrayProxy<int32_t> vals);
     virtual int writeToVector(ArrayProxy<int64_t> vals);
     virtual int writeToVector(ArrayProxy<unsigned char> vals);
@@ -357,12 +361,14 @@ public:
 
     virtual int setTensorValueDouble(const size_t bufLen, const double* vals);
     virtual int setTensorValueFloat(const size_t bufLen, const float* vals);
+    virtual int setTensorValueInt16(const size_t bufLen, const int16_t* vals);
     virtual int setTensorValueInt32(const size_t bufLen, const int32_t* vals);
     virtual int setTensorValueInt64(const size_t bufLen, const int64_t* vals);
     virtual int setTensorValueBool(const size_t bufLen, const bool* vals);
 
     virtual int getTensorValueDouble(const size_t bufLen, double* fbuf) const;
     virtual int getTensorValueFloat(const size_t bufLen, float* fbuf) const;
+    virtual int getTensorValueInt16(const size_t bufLen, int16_t* ibuf) const;
     virtual int getTensorValueInt32(const size_t bufLen, int32_t* ibuf) const;
     virtual int getTensorValueInt64(const size_t bufLen, int64_t* ibuf) const;
     virtual int getTensorValueBool(const size_t bufLen, bool* bbuf) const;
@@ -526,6 +532,21 @@ template <>
 int Tensor6<bool>::copyValueFrom(Tensor* src);
 
 template <>
+int Tensor0<int32_t>::setTensorValueInt16(const size_t bufLen, const int16_t* vals);
+template <>
+int Tensor1<int32_t>::setTensorValueInt16(const size_t bufLen, const int16_t* vals);
+template <>
+int Tensor2<int32_t>::setTensorValueInt16(const size_t bufLen, const int16_t* vals);
+template <>
+int Tensor3<int32_t>::setTensorValueInt16(const size_t bufLen, const int16_t* vals);
+template <>
+int Tensor4<int32_t>::setTensorValueInt16(const size_t bufLen, const int16_t* vals);
+template <>
+int Tensor5<int32_t>::setTensorValueInt16(const size_t bufLen, const int16_t* vals);
+template <>
+int Tensor6<int32_t>::setTensorValueInt16(const size_t bufLen, const int16_t* vals);
+
+template <>
 int Tensor0<int32_t>::setTensorValueInt32(const size_t bufLen, const int32_t* vals);
 template <>
 int Tensor1<int32_t>::setTensorValueInt32(const size_t bufLen, const int32_t* vals);
@@ -541,6 +562,21 @@ template <>
 int Tensor6<int32_t>::setTensorValueInt32(const size_t bufLen, const int32_t* vals);
 
 template <>
+int Tensor0<int32_t>::getTensorValueInt16(const size_t bufLen, int16_t* vals) const;
+template <>
+int Tensor1<int32_t>::getTensorValueInt16(const size_t bufLen, int16_t* vals) const;
+template <>
+int Tensor2<int32_t>::getTensorValueInt16(const size_t bufLen, int16_t* vals) const;
+template <>
+int Tensor3<int32_t>::getTensorValueInt16(const size_t bufLen, int16_t* vals) const;
+template <>
+int Tensor4<int32_t>::getTensorValueInt16(const size_t bufLen, int16_t* vals) const;
+template <>
+int Tensor5<int32_t>::getTensorValueInt16(const size_t bufLen, int16_t* vals) const;
+template <>
+int Tensor6<int32_t>::getTensorValueInt16(const size_t bufLen, int16_t* vals) const;
+
+template <>
 int Tensor0<int32_t>::getTensorValueInt32(const size_t bufLen, int32_t* vals) const;
 template <>
 int Tensor1<int32_t>::getTensorValueInt32(const size_t bufLen, int32_t* vals) const;