From ae2c5f0350a7033f58578f9c509345445a639865 Mon Sep 17 00:00:00 2001
From: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Date: Wed, 24 Apr 2019 16:19:57 +0100
Subject: IVGCVSW-2982 Refactor reference Activation workload

Change-Id: Ia3b9a56787cc68822a3c1635de82e03ecc0aae27
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
---
 src/backends/backendsCommon/WorkloadData.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'src/backends/backendsCommon')

diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index df66b4e4c1..66e30f0592 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -297,6 +297,20 @@ void ActivationQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
                               "ActivationQueueDescriptor",
                               "input",
                               "output");
+
+    std::vector<DataType> supportedTypes = {
+        DataType::Float32,
+        DataType::Float16,
+        DataType::QuantisedAsymm8
+    };
+
+    ValidateDataTypes(workloadInfo.m_InputTensorInfos[0],
+                      supportedTypes,
+                      "ActivationQueueDescriptor");
+
+    ValidateDataTypes(workloadInfo.m_OutputTensorInfos[0],
+                      {workloadInfo.m_InputTensorInfos[0].GetDataType()},
+                      "ActivationQueueDescriptor");
 }
 
 //---------------------------------------------------------------
-- 
cgit v1.2.1