From 043d0d0e07e0a29648894cb9ba68993e540bb94d Mon Sep 17 00:00:00 2001
From: Francis Murtagh <francis.murtagh@arm.com>
Date: Fri, 5 Oct 2018 14:08:48 +0100
Subject: IVGCVSW-1860 Support NHWC for Pooling2D

Change-Id: I9059a292e47867df82a5efbba5808fd264949ec9
---
 src/backends/test/LayerTests.cpp | 5 +++++
 1 file changed, 5 insertions(+)

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

diff --git a/src/backends/test/LayerTests.cpp b/src/backends/test/LayerTests.cpp
index a7fb6a824e..17f3ae12e1 100644
--- a/src/backends/test/LayerTests.cpp
+++ b/src/backends/test/LayerTests.cpp
@@ -4929,6 +4929,11 @@ LayerTestResult<float, 4> SimpleAveragePooling2dTest(armnn::IWorkloadFactory& wo
     return SimpleAveragePooling2dTestCommon<float>(workloadFactory);
 }
 
+LayerTestResult<float, 4> SimpleAveragePooling2dNhwcTest(armnn::IWorkloadFactory& workloadFactory)
+{
+    return SimpleAveragePooling2dNhwcTestCommon<float>(workloadFactory);
+}
+
 LayerTestResult<uint8_t, 4> SimpleAveragePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory)
 {
     return SimpleAveragePooling2dTestCommon<uint8_t>(workloadFactory, 0.5, -1);
-- 
cgit v1.2.1