From 5538d3436d6415d428b37d0f34ec81ca48cdff1a Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Wed, 14 Nov 2018 08:10:13 +0000 Subject: COMPMID-1781 Add channel support in CLL2Normalization Change-Id: Ibab049f09413258c99335b7da6b151530a1bd136 --- tests/validation/CL/L2NormalizeLayer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'tests/validation/CL/L2NormalizeLayer.cpp') diff --git a/tests/validation/CL/L2NormalizeLayer.cpp b/tests/validation/CL/L2NormalizeLayer.cpp index 517ba84069..fae7fd66db 100644 --- a/tests/validation/CL/L2NormalizeLayer.cpp +++ b/tests/validation/CL/L2NormalizeLayer.cpp @@ -46,8 +46,8 @@ namespace constexpr AbsoluteTolerance tolerance_f32(0.00001f); constexpr AbsoluteTolerance tolerance_f16(0.01f); -auto data = concat(combine(framework::dataset::make("DataLayout", { DataLayout::NCHW }), framework::dataset::make("Axis", { 0 })), combine(framework::dataset::make("DataLayout", { DataLayout::NHWC }), - framework::dataset::make("Axis", { 1 }))); +auto data = concat(combine(framework::dataset::make("DataLayout", { DataLayout::NCHW }), framework::dataset::make("Axis", { 0, 1, 2 })), combine(framework::dataset::make("DataLayout", { DataLayout::NHWC }), + framework::dataset::make("Axis", { 0, 1 }))); } // namespace -- cgit v1.2.1