From 735a450d3b53a2d745b9a7a6d85747e25ec37ede Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Wed, 26 Jun 2019 15:02:47 +0100 Subject: IVGCVSW-3320 Add reference workload support for TransposeConvolution2dLayer Signed-off-by: Aron Virginas-Tar Change-Id: Icc64f8148c9d8a0d14d772e6e4e7865e70585cd9 --- src/backends/reference/workloads/CMakeLists.txt | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) (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 daa004345c..9be245b1a7 100644 --- a/src/backends/reference/workloads/CMakeLists.txt +++ b/src/backends/reference/workloads/CMakeLists.txt @@ -29,6 +29,8 @@ list(APPEND armnnRefBackendWorkloads_sources Gather.hpp LstmUtils.hpp Maximum.hpp + Mean.cpp + Mean.hpp Concatenate.hpp Concatenate.cpp Minimum.hpp @@ -44,6 +46,8 @@ list(APPEND armnnRefBackendWorkloads_sources RefBatchNormalizationWorkload.hpp RefBatchToSpaceNdWorkload.cpp RefBatchToSpaceNdWorkload.hpp + RefConcatWorkload.cpp + RefConcatWorkload.hpp RefConstantWorkload.cpp RefConstantWorkload.hpp RefConvertFp16ToFp32Workload.cpp @@ -74,8 +78,8 @@ list(APPEND armnnRefBackendWorkloads_sources RefL2NormalizationWorkload.hpp RefLstmWorkload.cpp RefLstmWorkload.hpp - RefConcatWorkload.cpp - RefConcatWorkload.hpp + RefMeanWorkload.cpp + RefMeanWorkload.hpp RefNormalizationWorkload.cpp RefNormalizationWorkload.hpp RefPadWorkload.cpp @@ -104,6 +108,8 @@ list(APPEND armnnRefBackendWorkloads_sources RefSplitterWorkload.hpp RefStridedSliceWorkload.cpp RefStridedSliceWorkload.hpp + RefTransposeConvolution2dWorkload.cpp + RefTransposeConvolution2dWorkload.hpp RefWorkloads.hpp RefWorkloadUtils.hpp ResizeBilinear.cpp @@ -123,10 +129,8 @@ list(APPEND armnnRefBackendWorkloads_sources StringMapping.cpp StringMapping.hpp TensorBufferArrayView.hpp - Mean.cpp - Mean.hpp - RefMeanWorkload.cpp - RefMeanWorkload.hpp + TransposeConvolution2d.cpp + TransposeConvolution2d.hpp ) add_library(armnnRefBackendWorkloads OBJECT ${armnnRefBackendWorkloads_sources}) -- cgit v1.2.1