aboutsummaryrefslogtreecommitdiff
path: root/src/backends/test/LayerTests.hpp
diff options
context:
space:
mode:
authorjimfly01 <jim.flynn@arm.com>2018-10-31 14:43:53 +0000
committerMatteo Martincigh <matteo.martincigh@arm.com>2018-11-01 12:13:37 +0000
commit7af00da9068c71ae6f66eaa488048feea62e7aa9 (patch)
tree1c0808b0a1beeb85a3103f8fe01d561b1011ecc5 /src/backends/test/LayerTests.hpp
parent1b417c4970462d08e8e80d2f2462230ca5b91515 (diff)
downloadarmnn-7af00da9068c71ae6f66eaa488048feea62e7aa9.tar.gz
IVGCVSW-2074: Updated CompareDepthwiseConvolution2dTest for NHWC
Change-Id: Ib0568024b4f43ae17e6c75ac35d31cf103ce3482
Diffstat (limited to 'src/backends/test/LayerTests.hpp')
-rw-r--r--src/backends/test/LayerTests.hpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/test/LayerTests.hpp b/src/backends/test/LayerTests.hpp
index 392d3bf34e..006425ae5f 100644
--- a/src/backends/test/LayerTests.hpp
+++ b/src/backends/test/LayerTests.hpp
@@ -187,7 +187,8 @@ LayerTestResult<float, 4> CompareConvolution2dTest(armnn::IWorkloadFactory& work
template<typename T>
LayerTestResult<T, 4> CompareDepthwiseConvolution2dTest(armnn::IWorkloadFactory& workloadFactory,
- armnn::IWorkloadFactory& refWorkloadFactory);
+ armnn::IWorkloadFactory& refWorkloadFactory,
+ const armnn::DataLayoutIndexed& layout);
LayerTestResult<float, 4> CompareNormalizationTest(armnn::IWorkloadFactory& workloadFactory,
armnn::IWorkloadFactory& refWorkloadFactory,