aboutsummaryrefslogtreecommitdiff
path: root/src/backends/test/ArmComputeCl.cpp
diff options
context:
space:
mode:
authorFrancis Murtagh <francis.murtagh@arm.com>2018-10-04 16:03:07 +0100
committerMatthew Bentham <matthew.bentham@arm.com>2018-10-10 16:16:58 +0100
commitd59116ecb54c5bfe828d82ea0bc3367bc9b8c5dd (patch)
tree782d777544416eca8dcf8b924b710a395a75ad36 /src/backends/test/ArmComputeCl.cpp
parentde9011bc446d767932b6fec356f65791dff685e5 (diff)
downloadarmnn-d59116ecb54c5bfe828d82ea0bc3367bc9b8c5dd.tar.gz
IVGCVSW-1889 - Unit test Convolution2d with NHWC
* Added simple convolution Unit test * Set the data layout correctly in workloads Change-Id: Ie71b8415f6abc392a84900fc4438b7416fbb558a
Diffstat (limited to 'src/backends/test/ArmComputeCl.cpp')
-rw-r--r--src/backends/test/ArmComputeCl.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/test/ArmComputeCl.cpp b/src/backends/test/ArmComputeCl.cpp
index b4ec9bab16..d432a26f34 100644
--- a/src/backends/test/ArmComputeCl.cpp
+++ b/src/backends/test/ArmComputeCl.cpp
@@ -58,6 +58,8 @@ ARMNN_AUTO_TEST_CASE(UnbiasedConvolution2d, SimpleConvolution2d3x5Test, false)
ARMNN_AUTO_TEST_CASE(UnbiasedConvolution2dSquare, SimpleConvolution2d3x3Test, false)
ARMNN_AUTO_TEST_CASE(SimpleConvolution2dAsymmetricPadding, Convolution2dAsymmetricPaddingTest)
+ARMNN_AUTO_TEST_CASE(SimpleConvolution2dSquareNhwc, SimpleConvolution2d3x3NhwcTest, false)
+
// Depthwise Convolution
ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dDepthMul1, DepthwiseConvolution2dDepthMul1Test, true)
ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dDepthMul1, DepthwiseConvolution2dDepthMul1Test, false)