diff options
| author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-06-03 16:54:25 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-06-04 15:13:51 +0000 |
| commit | 3122bd574a3d29774c535ca2136de361da626e88 (patch) | |
| tree | c2fcc19be67f5a35c30d042b80ba3157ef87bd21 /src/backends/reference/workloads/CMakeLists.txt | |
| parent | 550fe36f687e73c78b57ebfeee9f98fd35f40f24 (diff) | |
| download | armnn-3122bd574a3d29774c535ca2136de361da626e88.tar.gz | |
IVGCVSW-3212 Refactor the Reference BatchNormalization workloads to
handle Float32 and QAsymm8 types
* Removed the type-specific workload implementations
* Added type-independent RefBatchNormalizationWorkload implementation
* Reworked BachNormImpl to use decoders/encoders
* Improved the validation of the BatchNorm queue descriptor
* Fixed unit tests where necessary
Change-Id: Icf3fa1332292d38ec2fa0b1cb984cab78426034b
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Diffstat (limited to 'src/backends/reference/workloads/CMakeLists.txt')
| -rw-r--r-- | src/backends/reference/workloads/CMakeLists.txt | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/CMakeLists.txt b/src/backends/reference/workloads/CMakeLists.txt index df126c4308..cdca22da31 100644 --- a/src/backends/reference/workloads/CMakeLists.txt +++ b/src/backends/reference/workloads/CMakeLists.txt @@ -7,6 +7,7 @@ list(APPEND armnnRefBackendWorkloads_sources Activation.cpp Activation.hpp BaseIterator.hpp + BatchNormImpl.cpp BatchNormImpl.hpp BatchToSpaceNd.cpp BatchToSpaceNd.hpp @@ -37,10 +38,8 @@ list(APPEND armnnRefBackendWorkloads_sources Pooling2d.hpp RefActivationWorkload.cpp RefActivationWorkload.hpp - RefBatchNormalizationFloat32Workload.cpp - RefBatchNormalizationFloat32Workload.hpp - RefBatchNormalizationUint8Workload.cpp - RefBatchNormalizationUint8Workload.hpp + RefBatchNormalizationWorkload.cpp + RefBatchNormalizationWorkload.hpp RefBatchToSpaceNdFloat32Workload.cpp RefBatchToSpaceNdFloat32Workload.hpp RefBatchToSpaceNdUint8Workload.cpp |