diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2020-12-08 21:02:16 +0000 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-01-05 14:30:17 +0000 |
| commit | a3c9a3b3d56f0369b199512fef832e6db958a601 (patch) | |
| tree | 357bf1ea0c3ccf2ac314b0777036642a11b5f7cd /src/core/NEON/kernels/NEArithmeticAdditionKernel.h | |
| parent | b309fc249e4383b4d40ae03e377c3cbad3f9f5f7 (diff) | |
| download | ComputeLibrary-a3c9a3b3d56f0369b199512fef832e6db958a601.tar.gz | |
COMPMID-3874: Create ArithmeticAddition SVE/SVE2
Change-Id: I4ec7561a7f6a42a22b8187968ae302dbe75023bc
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4753
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NEArithmeticAdditionKernel.h')
| -rw-r--r-- | src/core/NEON/kernels/NEArithmeticAdditionKernel.h | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/src/core/NEON/kernels/NEArithmeticAdditionKernel.h b/src/core/NEON/kernels/NEArithmeticAdditionKernel.h index 2072ad91bd..b88fc8aa74 100644 --- a/src/core/NEON/kernels/NEArithmeticAdditionKernel.h +++ b/src/core/NEON/kernels/NEArithmeticAdditionKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2016-2020 Arm Limited. + * Copyright (c) 2016-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -88,7 +88,6 @@ public: // Inherited methods overridden: void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override; -private: /** Common signature for all the specialised add functions * * @param[in] input1 First input tensor. Data types supported: U8/QASYMM8/S16/QSYMM16/F16/S32/F32 @@ -97,10 +96,12 @@ private: * @param[in] policy Overflow policy. * @param[in] window Region on which to execute the kernel. */ - using AddFunction = void(const ITensor *input1, const ITensor *input2, ITensor *output, ConvertPolicy policy, const Window &window); + using ArithmeticAdditionKernelPtr = std::add_pointer<void(const ITensor *, const ITensor *, ITensor *, const ConvertPolicy &, const Window &)>::type; + +private: /** Add function to use for the particular tensor types passed to configure() */ - AddFunction *_func; - ConvertPolicy _policy; + ArithmeticAdditionKernelPtr _func; + ConvertPolicy _policy; }; } // namespace arm_compute #endif /*ARM_COMPUTE_NEARITHMETICADDITIONKERNEL_H */ |