commit cbd2c230b7ce5f26e2ccccf36b7ad450f6e1ad09
author Finn Williams <Finn.Williams@arm.com> Mon Jun 22 15:58:32 2020 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Tue Jun 30 13:00:52 2020 +0000
tree eb6e5393726be21213e72f26a676b7c3809fc995
parent 532a29d12d72f54549d8b71edd485c17af65698a

IVGCVSW-5007 Implement an Int32 reference Elementwise workload

Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I6592169b74ac4294bc09647879aec0718c641f91

src/backends/reference/workloads/Decoders.hpp

diff --git a/src/backends/reference/workloads/Decoders.hpp b/src/backends/reference/workloads/Decoders.hpp
index deb3b1f..08e0140 100644
--- a/src/backends/reference/workloads/Decoders.hpp
+++ b/src/backends/reference/workloads/Decoders.hpp
@@ -149,4 +149,22 @@
     return nullptr;
 }
 
+template<>
+inline std::unique_ptr<Decoder<int32_t>> MakeDecoder(const TensorInfo& info, const void* data)
+{
+    switch(info.GetDataType())
+    {
+        case DataType::Signed32:
+        {
+            return std::make_unique<Int32ToInt32tDecoder>(static_cast<const int32_t*>(data));
+        }
+        default:
+        {
+            ARMNN_ASSERT_MSG(false, "Unsupported Data Type!");
+            break;
+        }
+    }
+    return nullptr;
+}
+
 } //namespace armnn