aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/CMakeLists.txt
diff options
context:
space:
mode:
authorÉanna Ó Catháin <eanna.ocathain@arm.com>2018-11-12 11:36:34 +0000
committerLes Bell <les.bell@arm.com>2018-11-12 13:08:37 +0000
commit4e1e136cce3fca73ba49b570cfcb620f4ec574da (patch)
tree1fe9fcbb6a9dbafc12aa99ac543bc0da636a1cd1 /src/backends/reference/workloads/CMakeLists.txt
parentf97debb95cbc7e0bbc60e66e5463ede517cac61b (diff)
downloadarmnn-4e1e136cce3fca73ba49b570cfcb620f4ec574da.tar.gz
IVGCVSW-2054: BATCH_TO_SPACE_ND Reference implementation and Unit tests.
Change-Id: I13c6728dbb60643d0e086d171225c5d802987f92
Diffstat (limited to 'src/backends/reference/workloads/CMakeLists.txt')
-rw-r--r--src/backends/reference/workloads/CMakeLists.txt6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/CMakeLists.txt b/src/backends/reference/workloads/CMakeLists.txt
index 4cef2d0771..1c38509ca0 100644
--- a/src/backends/reference/workloads/CMakeLists.txt
+++ b/src/backends/reference/workloads/CMakeLists.txt
@@ -9,6 +9,8 @@ list(APPEND armnnRefBackendWorkloads_sources
ArithmeticFunction.cpp
ArithmeticFunction.hpp
BatchNormImpl.hpp
+ BatchToSpaceNd.cpp
+ BatchToSpaceNd.hpp
Broadcast.cpp
Broadcast.hpp
ConvImpl.cpp
@@ -32,6 +34,10 @@ list(APPEND armnnRefBackendWorkloads_sources
RefBatchNormalizationFloat32Workload.hpp
RefBatchNormalizationUint8Workload.cpp
RefBatchNormalizationUint8Workload.hpp
+ RefBatchToSpaceNdFloat32Workload.cpp
+ RefBatchToSpaceNdFloat32Workload.hpp
+ RefBatchToSpaceNdUint8Workload.cpp
+ RefBatchToSpaceNdUint8Workload.hpp
RefConstantFloat32Workload.cpp
RefConstantFloat32Workload.hpp
RefConstantUint8Workload.cpp