aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/NEON/kernels/NENormalizationLayerKernel.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/NEON/kernels/NENormalizationLayerKernel.h')
-rw-r--r--arm_compute/core/NEON/kernels/NENormalizationLayerKernel.h4
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/core/NEON/kernels/NENormalizationLayerKernel.h b/arm_compute/core/NEON/kernels/NENormalizationLayerKernel.h
index d4e36d5ff1..b1bc594e4c 100644
--- a/arm_compute/core/NEON/kernels/NENormalizationLayerKernel.h
+++ b/arm_compute/core/NEON/kernels/NENormalizationLayerKernel.h
@@ -73,8 +73,8 @@ private:
*
* @param[in] window Region on which to execute the kernel.
*/
- template <unsigned int dim, bool do_2D_norm>
- void normalize(const Window &window);
+ template <DataType dt, unsigned int dim, bool do_2D_norm>
+ void normalize_float(const Window &window);
/** Function to perform normalization for fixed-point values depending on
* the given template dimension. The second template parameter specifies