aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/test/NeonLayerTests.cpp
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/neon/test/NeonLayerTests.cpp
parent1b417c4970462d08e8e80d2f2462230ca5b91515 (diff)
downloadarmnn-7af00da9068c71ae6f66eaa488048feea62e7aa9.tar.gz
IVGCVSW-2074: Updated CompareDepthwiseConvolution2dTest for NHWC
Change-Id: Ib0568024b4f43ae17e6c75ac35d31cf103ce3482
Diffstat (limited to 'src/backends/neon/test/NeonLayerTests.cpp')
-rw-r--r--src/backends/neon/test/NeonLayerTests.cpp15
1 files changed, 13 insertions, 2 deletions
diff --git a/src/backends/neon/test/NeonLayerTests.cpp b/src/backends/neon/test/NeonLayerTests.cpp
index d242245ba0..8fda825618 100644
--- a/src/backends/neon/test/NeonLayerTests.cpp
+++ b/src/backends/neon/test/NeonLayerTests.cpp
@@ -439,8 +439,19 @@ ARMNN_AUTO_TEST_CASE(SimpleNormalizationAcrossNhwc, SimpleNormalizationAcrossNhw
ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareConv2dWithReference, CompareConvolution2dTest)
-ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceFloat32, CompareDepthwiseConvolution2dTest<float>)
-ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceUint8, CompareDepthwiseConvolution2dTest<uint8_t>)
+ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceFloat32,
+ CompareDepthwiseConvolution2dTest<float>,
+ armnn::DataLayout::NCHW)
+ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceUint8,
+ CompareDepthwiseConvolution2dTest<uint8_t>,
+ armnn::DataLayout::NCHW)
+
+ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceFloat32Nhwc,
+ CompareDepthwiseConvolution2dTest<float>,
+ armnn::DataLayout::NHWC)
+ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceUint8Nhwc,
+ CompareDepthwiseConvolution2dTest<uint8_t>,
+ armnn::DataLayout::NHWC)
ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareNormalizationWithinWithReference, CompareNormalizationTest,
armnn::NormalizationAlgorithmChannel::Within,