diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-08-22 16:20:21 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
commit | aaba4c626bcc6365e0108130633ce43fafe9da45 (patch) | |
tree | 768ee62e90b9bf39dfbfe3a6551a45071e83c6c3 /arm_compute/core/NEON/kernels | |
parent | 630d941836e32fadca62549c177cb57d4436b44b (diff) | |
download | ComputeLibrary-aaba4c626bcc6365e0108130633ce43fafe9da45.tar.gz |
COMPMID-1188: Add support for activation in NEBatchNormalization.
Change-Id: I1e206574dac6433218db6e138adb7bf5f66a536d
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/145222
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/kernels')
-rw-r--r-- | arm_compute/core/NEON/kernels/NEBatchNormalizationLayerKernel.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/core/NEON/kernels/NEBatchNormalizationLayerKernel.h b/arm_compute/core/NEON/kernels/NEBatchNormalizationLayerKernel.h index 2a540c151b..092b1d9514 100644 --- a/arm_compute/core/NEON/kernels/NEBatchNormalizationLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NEBatchNormalizationLayerKernel.h @@ -103,7 +103,7 @@ private: * * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()). */ - template <bool fused_activation> + template <bool fused_activation, typename F> void batch_normalization_fp16_nchw(const Window &window); /** Template function to run batch normalization on fp16 on tensors with NHWC format * @@ -111,7 +111,7 @@ private: * * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()). */ - template <bool fused_activation> + template <bool fused_activation, typename F> void batch_normalization_fp16_nhwc(const Window &window); /** Template function to run batch normalization on fp32 * |