From cec6b655d9f6ddb73c721ca47a7d67eeaad5c043 Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Fri, 12 Oct 2018 13:51:57 +0100 Subject: IVGCVSW-1922 Unit test for DepthwiseConvolution with NHWC Change-Id: I3e6e5b9a62f30d03c05bd7178adea8f4c8275da8 --- src/backends/test/LayerTests.hpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/backends/test/LayerTests.hpp') diff --git a/src/backends/test/LayerTests.hpp b/src/backends/test/LayerTests.hpp index f5abd985c8..9f8cd3ff25 100644 --- a/src/backends/test/LayerTests.hpp +++ b/src/backends/test/LayerTests.hpp @@ -68,6 +68,9 @@ LayerTestResult Convolution1dUint8Test(armnn::IWorkloadFactory& work LayerTestResult DepthwiseConvolution2dTest(armnn::IWorkloadFactory& workloadFactory, bool biasEnabled); +LayerTestResult DepthwiseConvolution2dDepthNhwcTest(armnn::IWorkloadFactory& workloadFactory, + bool biasEnabled); + LayerTestResult DepthwiseConvolution2dDepthMul1Test(armnn::IWorkloadFactory& workloadFactory, bool biasEnabled); -- cgit v1.2.1