diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2020-07-02 17:39:25 +0100 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2020-07-07 09:14:00 +0000 |
| commit | 6eb73458c4869165c88d33c6a745a91cdc73a36a (patch) | |
| tree | 1f22bd141f420ad4e2906939bb4abf11fec3aea3 /arm_compute/runtime/NEON/functions/NENormalizationLayer.h | |
| parent | 1fad814022ba98506ba30b2e25601985e7ec5259 (diff) | |
| download | ComputeLibrary-6eb73458c4869165c88d33c6a745a91cdc73a36a.tar.gz | |
COMPMID-3373: Async support to NEArithmetic* kernels/functions (Pt. 2)
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Change-Id: Iec06adb535aaf7efb1838d921e8d6bb978b7b215
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3498
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NENormalizationLayer.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NENormalizationLayer.h | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/arm_compute/runtime/NEON/functions/NENormalizationLayer.h b/arm_compute/runtime/NEON/functions/NENormalizationLayer.h index 8683e44d3c..bead01457f 100644 --- a/arm_compute/runtime/NEON/functions/NENormalizationLayer.h +++ b/arm_compute/runtime/NEON/functions/NENormalizationLayer.h @@ -28,10 +28,10 @@ #include "arm_compute/core/NEON/kernels/NEFillBorderKernel.h" #include "arm_compute/core/NEON/kernels/NENormalizationLayerKernel.h" -#include "arm_compute/core/NEON/kernels/NEPixelWiseMultiplicationKernel.h" #include "arm_compute/core/Types.h" #include "arm_compute/runtime/IMemoryManager.h" #include "arm_compute/runtime/MemoryGroup.h" +#include "arm_compute/runtime/NEON/functions/NEPixelWiseMultiplication.h" #include "arm_compute/runtime/Tensor.h" #include <memory> @@ -42,7 +42,7 @@ class ITensor; /** Basic function to compute a normalization layer. This function calls the following NEON kernels: * - * -# @ref NEPixelWiseMultiplicationKernel + * -# @ref NEPixelWiseMultiplication * -# @ref NEFillBorderKernel * -# @ref NENormalizationLayerKernel * @@ -75,10 +75,10 @@ public: void run() override; private: - MemoryGroup _memory_group; /**< Function memory group */ - NENormalizationLayerKernel _norm_kernel; /**< Normalization layer kernel */ - NEPixelWiseMultiplicationKernel _multiply_kernel; /**< Pixel multiplication kernel */ - Tensor _input_squared; /**< The intermediate buffer which stores results of squaring input */ + MemoryGroup _memory_group; /**< Function memory group */ + NENormalizationLayerKernel _norm_kernel; /**< Normalization layer kernel */ + NEPixelWiseMultiplication _multiply_f; /**< Pixel multiplication function */ + Tensor _input_squared; /**< The intermediate buffer which stores results of squaring input */ }; } #endif /* ARM_COMPUTE_NENORMALIZATIONLAYER_H */ |