diff options
Diffstat (limited to 'src/armnn/backends/NeonWorkloads/NeonL2NormalizationFloat32Workload.hpp')
-rw-r--r-- | src/armnn/backends/NeonWorkloads/NeonL2NormalizationFloat32Workload.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnn/backends/NeonWorkloads/NeonL2NormalizationFloat32Workload.hpp b/src/armnn/backends/NeonWorkloads/NeonL2NormalizationFloat32Workload.hpp index 2b4a1fef37..c3fcde5a57 100644 --- a/src/armnn/backends/NeonWorkloads/NeonL2NormalizationFloat32Workload.hpp +++ b/src/armnn/backends/NeonWorkloads/NeonL2NormalizationFloat32Workload.hpp @@ -14,7 +14,10 @@ namespace armnn { -class NeonL2NormalizationFloat32Workload : public Float32Workload<L2NormalizationQueueDescriptor> +arm_compute::Status NeonL2NormalizationWorkloadValidate(const TensorInfo& input, + const TensorInfo& output); + +class NeonL2NormalizationFloat32Workload : public FloatWorkload<L2NormalizationQueueDescriptor> { public: NeonL2NormalizationFloat32Workload(const L2NormalizationQueueDescriptor& descriptor, const WorkloadInfo& info, |