From d134093a271b60e248942af9757e8236e8f41ac1 Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Thu, 18 Oct 2018 14:27:50 +0100 Subject: IVGCVSW-2023 CL and Neon implementation of BatchNorm with NHWC Change-Id: I962e986607e5d045cd97b9eaeaea2f5ae624db35 --- src/backends/reference/test/RefCreateWorkloadTests.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/backends/reference/test') diff --git a/src/backends/reference/test/RefCreateWorkloadTests.cpp b/src/backends/reference/test/RefCreateWorkloadTests.cpp index d258b81932..8bad5497a2 100644 --- a/src/backends/reference/test/RefCreateWorkloadTests.cpp +++ b/src/backends/reference/test/RefCreateWorkloadTests.cpp @@ -157,13 +157,13 @@ static void RefCreateBatchNormalizationWorkloadTest(DataLayout dataLayout) switch (dataLayout) { case DataLayout::NHWC: - inputShape = { 2, 1, 1, 3 }; - outputShape = { 2, 1, 1, 3 }; + inputShape = { 2, 4, 4, 3 }; + outputShape = { 2, 4, 4, 3 }; break; case DataLayout::NCHW: default: - inputShape = { 2, 3, 1, 1 }; - outputShape = { 2, 3, 1, 1 }; + inputShape = { 2, 3, 4, 4 }; + outputShape = { 2, 3, 4, 4 }; break; } -- cgit v1.2.1