diff options
| author | narpra01 <narumol.prangnawarat@arm.com> | 2018-10-02 14:35:53 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:58 +0100 |
| commit | 55a97bc2605fc1246a9a1f7ee89cde415496a1ba (patch) | |
| tree | 28043aa8cbe684f978d46c690b100000e9517312 /src/backends/test/CreateWorkloadRef.cpp | |
| parent | ee9e7665a5922f7ec0c5ec24d6ab2ecd88fbcfd6 (diff) | |
| download | armnn-55a97bc2605fc1246a9a1f7ee89cde415496a1ba.tar.gz | |
IVGCVSW-1920 Unittests for NHWC Normalization Workloads and Layer
Change-Id: Iea941c1747454f5a4342351e4e82b10ffb9ccbbd
Diffstat (limited to 'src/backends/test/CreateWorkloadRef.cpp')
| -rw-r--r-- | src/backends/test/CreateWorkloadRef.cpp | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/src/backends/test/CreateWorkloadRef.cpp b/src/backends/test/CreateWorkloadRef.cpp index 9313ee851f..c30093da92 100644 --- a/src/backends/test/CreateWorkloadRef.cpp +++ b/src/backends/test/CreateWorkloadRef.cpp @@ -227,17 +227,22 @@ BOOST_AUTO_TEST_CASE(CreateFullyConnectedUint8Workload) RefCreateFullyConnectedWorkloadTest<RefFullyConnectedUint8Workload, armnn::DataType::QuantisedAsymm8>(); } -BOOST_AUTO_TEST_CASE(CreateNormalizationWorkload) +template <typename NormalizationWorkloadType, armnn::DataType DataType> +static void RefCreateNormalizationWorkloadTest() { - Graph graph; + Graph graph; RefWorkloadFactory factory; - auto workload = CreateNormalizationWorkloadTest<RefNormalizationFloat32Workload, - armnn::DataType::Float32>(factory, graph); + auto workload = CreateNormalizationWorkloadTest<NormalizationWorkloadType, DataType>(factory, graph); // Checks that outputs and inputs are as we expect them (see definition of CreateNormalizationWorkloadTest). CheckInputOutput(std::move(workload), - TensorInfo({3, 5, 5, 1}, DataType::Float32), - TensorInfo({3, 5, 5, 1}, DataType::Float32)); + TensorInfo({3, 5, 5, 1}, DataType), + TensorInfo({3, 5, 5, 1}, DataType)); +} + +BOOST_AUTO_TEST_CASE(CreateRefNormalizationNchwWorkload) +{ + RefCreateNormalizationWorkloadTest<RefNormalizationFloat32Workload, armnn::DataType::Float32>(); } template <typename Pooling2dWorkloadType, armnn::DataType DataType> |