aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/CMakeLists.txt
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/CMakeLists.txt')
-rw-r--r--src/backends/reference/workloads/CMakeLists.txt4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/CMakeLists.txt b/src/backends/reference/workloads/CMakeLists.txt
index 7dc72891c7..4cef2d0771 100644
--- a/src/backends/reference/workloads/CMakeLists.txt
+++ b/src/backends/reference/workloads/CMakeLists.txt
@@ -86,6 +86,8 @@ list(APPEND armnnRefBackendWorkloads_sources
RefSoftmaxFloat32Workload.hpp
RefSoftmaxUint8Workload.cpp
RefSoftmaxUint8Workload.hpp
+ RefSpaceToBatchNdWorkload.cpp
+ RefSpaceToBatchNdWorkload.hpp
RefSplitterFloat32Workload.cpp
RefSplitterFloat32Workload.hpp
RefSplitterUint8Workload.cpp
@@ -96,6 +98,8 @@ list(APPEND armnnRefBackendWorkloads_sources
ResizeBilinear.hpp
Softmax.cpp
Softmax.hpp
+ SpaceToBatchNd.hpp
+ SpaceToBatchNd.cpp
Splitter.hpp
TensorBufferArrayView.hpp
Mean.cpp