commit 55a97bc2605fc1246a9a1f7ee89cde415496a1ba
author narpra01 <narumol.prangnawarat@arm.com> Tue Oct 02 14:35:53 2018 +0100
committer Matthew Bentham <matthew.bentham@arm.com> Wed Oct 10 16:16:58 2018 +0100
tree 28043aa8cbe684f978d46c690b100000e9517312
parent ee9e7665a5922f7ec0c5ec24d6ab2ecd88fbcfd6

IVGCVSW-1920 Unittests for NHWC Normalization Workloads and Layer

Change-Id: Iea941c1747454f5a4342351e4e82b10ffb9ccbbd

src/backends/cl/workloads/ClNormalizationFloatWorkload.cpp

diff --git a/src/backends/cl/workloads/ClNormalizationFloatWorkload.cpp b/src/backends/cl/workloads/ClNormalizationFloatWorkload.cpp
index 969c9bb..f6c07e1 100644
--- a/src/backends/cl/workloads/ClNormalizationFloatWorkload.cpp
+++ b/src/backends/cl/workloads/ClNormalizationFloatWorkload.cpp
@@ -37,6 +37,10 @@
     arm_compute::ICLTensor& input  = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
     arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
 
+    arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
+    input.info()->set_data_layout(aclDataLayout);
+    output.info()->set_data_layout(aclDataLayout);
+
     arm_compute::NormalizationLayerInfo normalizationInfo = BuildArmComputeNormalizationLayerInfo(m_Data.m_Parameters);
 
     m_NormalizationLayer.configure(&input, &output, normalizationInfo);