diff options
Diffstat (limited to 'src/backends/ClWorkloads')
-rw-r--r-- | src/backends/ClWorkloads/ClL2NormalizationFloatWorkload.cpp | 7 | ||||
-rw-r--r-- | src/backends/ClWorkloads/ClL2NormalizationFloatWorkload.hpp | 3 |
2 files changed, 6 insertions, 4 deletions
diff --git a/src/backends/ClWorkloads/ClL2NormalizationFloatWorkload.cpp b/src/backends/ClWorkloads/ClL2NormalizationFloatWorkload.cpp index 8f5ed5feca..0ca334348e 100644 --- a/src/backends/ClWorkloads/ClL2NormalizationFloatWorkload.cpp +++ b/src/backends/ClWorkloads/ClL2NormalizationFloatWorkload.cpp @@ -15,10 +15,11 @@ namespace armnn using namespace armcomputetensorutils; arm_compute::Status ClL2NormalizationWorkloadValidate(const TensorInfo& input, - const TensorInfo& output) + const TensorInfo& output, + const L2NormalizationDescriptor& descriptor) { - const arm_compute::TensorInfo aclInput = BuildArmComputeTensorInfo(input); - const arm_compute::TensorInfo aclOutput = BuildArmComputeTensorInfo(output); + const arm_compute::TensorInfo aclInput = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); + const arm_compute::TensorInfo aclOutput = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); arm_compute::NormalizationLayerInfo normalizationInfo = CreateAclNormalizationLayerInfoForL2Normalization(input); diff --git a/src/backends/ClWorkloads/ClL2NormalizationFloatWorkload.hpp b/src/backends/ClWorkloads/ClL2NormalizationFloatWorkload.hpp index f3f7de110a..20c0426208 100644 --- a/src/backends/ClWorkloads/ClL2NormalizationFloatWorkload.hpp +++ b/src/backends/ClWorkloads/ClL2NormalizationFloatWorkload.hpp @@ -13,7 +13,8 @@ namespace armnn { arm_compute::Status ClL2NormalizationWorkloadValidate(const TensorInfo& input, - const TensorInfo& output); + const TensorInfo& output, + const L2NormalizationDescriptor& descriptor); class ClL2NormalizationFloatWorkload : public FloatWorkload<L2NormalizationQueueDescriptor> { |