aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/test/RefLayerTests.cpp
diff options
context:
space:
mode:
authorjimfly01 <jim.flynn@arm.com>2018-10-26 15:55:50 +0100
committerLes Bell <les.bell@arm.com>2018-10-29 14:54:10 +0000
commit382a91d5029e83002bda4ab006f9c73340d679fe (patch)
tree0f96de71540963208a028806c4290d87ece2f9ce /src/backends/reference/test/RefLayerTests.cpp
parent19483db955cfa570d04d0fdf65bf8fad161ea3e9 (diff)
downloadarmnn-382a91d5029e83002bda4ab006f9c73340d679fe.tar.gz
IVGCVSW-2074: Updated DepthwiseConvolution2dAsymmetricTest for NHWC
Change-Id: I1bc8d2f270d4abfaa62e4843b8d96bcf462c6b09
Diffstat (limited to 'src/backends/reference/test/RefLayerTests.cpp')
-rw-r--r--src/backends/reference/test/RefLayerTests.cpp10
1 files changed, 8 insertions, 2 deletions
diff --git a/src/backends/reference/test/RefLayerTests.cpp b/src/backends/reference/test/RefLayerTests.cpp
index cb5a1c4158..6a92b67e57 100644
--- a/src/backends/reference/test/RefLayerTests.cpp
+++ b/src/backends/reference/test/RefLayerTests.cpp
@@ -67,8 +67,14 @@ ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dDepthMul1Uint8, DepthwiseConvolution2
ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dDepthMul1, DepthwiseConvolution2dDepthMul1Test, false)
ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dDepthMul1Uint8, DepthwiseConvolution2dDepthMul1Uint8Test, false)
-ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dAsymmetric, DepthwiseConvolution2dAsymmetricTest, true)
-ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dAsymmetric, DepthwiseConvolution2dAsymmetricTest, false)
+ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dAsymmetric,
+ DepthwiseConvolution2dAsymmetricTest, true, armnn::DataLayout::NCHW)
+ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dAsymmetric,
+ DepthwiseConvolution2dAsymmetricTest, false, armnn::DataLayout::NCHW)
+ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dAsymmetricNhwc,
+ DepthwiseConvolution2dAsymmetricTest, true, armnn::DataLayout::NHWC)
+ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dAsymmetricNhwc,
+ DepthwiseConvolution2dAsymmetricTest, false, armnn::DataLayout::NHWC)
// Pooling
ARMNN_AUTO_TEST_CASE(SimpleMaxPooling2dSize2x2Stride2x2, SimpleMaxPooling2dSize2x2Stride2x2Test, false)