From 7af00da9068c71ae6f66eaa488048feea62e7aa9 Mon Sep 17 00:00:00 2001
From: jimfly01 <jim.flynn@arm.com>
Date: Wed, 31 Oct 2018 14:43:53 +0000
Subject: IVGCVSW-2074: Updated CompareDepthwiseConvolution2dTest for NHWC

Change-Id: Ib0568024b4f43ae17e6c75ac35d31cf103ce3482
---
 src/backends/test/LayerTests.hpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src/backends/test/LayerTests.hpp')

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,
-- 
cgit v1.2.1