aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/WorkloadData.cpp
diff options
context:
space:
mode:
authorJames Conroy <james.conroy@arm.com>2019-05-30 16:36:59 +0100
committerJames Conroy <james.conroy@arm.com>2019-05-31 15:56:38 +0100
commit83735b177a4c71952d36e7a7919537b5a4d6d49e (patch)
tree51a2e30b1f6caeea9994848c3be6f93471d03b64 /src/backends/backendsCommon/WorkloadData.cpp
parent983daec95bd3b3e5558771e995092981f2df2373 (diff)
downloadarmnn-83735b177a4c71952d36e7a7919537b5a4d6d49e.tar.gz
IVGCVSW-3177 Refactor Floor reference workload
* Renamed RefFloorFloat32Workload to RefFloorWorkload and updated references to reflect this change. * RefFloorWorkload now uses Decoders/Encoders and supports the use of multiple data types. * Deleted FloorTestImpl.hpp and moved its contents into LayerTests.hpp. Change-Id: Ie079d05f2f6a578172f0fe3024f9607c030bce64 Signed-off-by: James Conroy <james.conroy@arm.com>
Diffstat (limited to 'src/backends/backendsCommon/WorkloadData.cpp')
-rw-r--r--src/backends/backendsCommon/WorkloadData.cpp16
1 files changed, 13 insertions, 3 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index d2e893f102..9cc8ba07a8 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -1024,12 +1024,22 @@ void SpaceToBatchNdQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) c
void FloorQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
{
- ValidateNumInputs(workloadInfo, "FloorQueueDescriptor", 1);
- ValidateNumOutputs(workloadInfo, "FlootQueueDescriptor", 1);
+ const std::string floorQueueDescString = "FloorQueueDescriptor";
+
+ ValidateNumInputs(workloadInfo, floorQueueDescString, 1);
+ ValidateNumOutputs(workloadInfo, floorQueueDescString, 1);
+
+ std::vector<DataType> supportedTypes =
+ {
+ DataType::Float32
+ };
+
+ ValidateDataTypes(workloadInfo.m_InputTensorInfos[0], supportedTypes, floorQueueDescString);
+ ValidateDataTypes(workloadInfo.m_OutputTensorInfos[0], supportedTypes, floorQueueDescString);
if (workloadInfo.m_InputTensorInfos[0] != workloadInfo.m_OutputTensorInfos[0])
{
- throw InvalidArgumentException("FloorQueueDescriptor: Input and output tensor infos do not match.");
+ throw InvalidArgumentException(floorQueueDescString + ": Input and output tensor infos do not match.");
}
}