From 3ea76d5f0d99794cf5f0b60ef3738d0905f10b2a Mon Sep 17 00:00:00 2001
From: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Date: Fri, 9 Nov 2018 14:10:38 +0000
Subject: IVGCVSW-2095 Add reference implementation and unit tests for
 SpaceToBatchNd

Change-Id: I27ffebdece6e68460931a44c15b9b029f9fce638
---
 src/backends/reference/workloads/CMakeLists.txt | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'src/backends/reference/workloads/CMakeLists.txt')

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
-- 
cgit v1.2.1