commit 5edc8816118fcddb2681379db04c978041ce8b46
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Tue Nov 05 18:00:21 2019 +0000
committer Francis Murtagh <francis.murtagh@arm.com> Wed Nov 06 12:10:02 2019 +0000
tree 22e4382138e9963d0ed3dacefda4fb142877e1fc
parent ec33a91ec1557b78b2d01975ec4c5eaf24aa058c

IVGCVSW-3837 Add support for per-axis quantization to reference Convolution2d workload

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I0ac08ba4864d48e6f64c4ac645dad8ea850be112

src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp

diff --git a/src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp
index f5ff586..3aac975 100644
--- a/src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp
+++ b/src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp
@@ -111,6 +111,11 @@
     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
     armnn::IWorkloadFactory& refWorkloadFactory);
 
+LayerTestResult<uint8_t, 4> Convolution2dPerAxisQuantTest(
+    armnn::IWorkloadFactory& workloadFactory,
+    const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
+    const armnn::DataLayout layout);
+
 //
 // DepthwiseConvolution2d
 //