aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefWorkloads.hpp
diff options
context:
space:
mode:
authorFrancis Murtagh <francis.murtagh@arm.com>2019-06-20 12:07:19 +0100
committerFrancis Murtagh <francis.murtagh@arm.com>2019-06-20 12:07:30 +0100
commit47ea3c0e8d8d10906d04a0e7c537ffee68b0f819 (patch)
treeab160c7126820dd2b5766974256167fdc66d183a /src/backends/reference/workloads/RefWorkloads.hpp
parent51982472bfedf12e7d82cde6614617f94b2c86d0 (diff)
downloadarmnn-47ea3c0e8d8d10906d04a0e7c537ffee68b0f819.tar.gz
IVGCVSW-3248 Refactor reference BatchToSpace workload
* Add Decoders and Encoders to workload to make it data type agnostic * Merge float32 and Uint8 into single workload Change-Id: I8adfa1898a63f13889eaaf55a31c26fd1e2d7ee8 Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/backends/reference/workloads/RefWorkloads.hpp')
-rw-r--r--src/backends/reference/workloads/RefWorkloads.hpp3
1 files changed, 1 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/RefWorkloads.hpp b/src/backends/reference/workloads/RefWorkloads.hpp
index 056127aae6..90582812e7 100644
--- a/src/backends/reference/workloads/RefWorkloads.hpp
+++ b/src/backends/reference/workloads/RefWorkloads.hpp
@@ -15,8 +15,7 @@
#include "Pooling2d.hpp"
#include "RefActivationWorkload.hpp"
#include "RefBatchNormalizationWorkload.hpp"
-#include "RefBatchToSpaceNdUint8Workload.hpp"
-#include "RefBatchToSpaceNdFloat32Workload.hpp"
+#include "RefBatchToSpaceNdWorkload.hpp"
#include "RefConvolution2dWorkload.hpp"
#include "RefConstantWorkload.hpp"
#include "RefConcatWorkload.hpp"