diff options
| author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2021-02-03 10:32:59 +0000 |
|---|---|---|
| committer | Sang-Hoon Park <sang-hoon.park@arm.com> | 2021-03-08 17:00:45 +0000 |
| commit | 668ccdcfb81bfab3a2d44cd1ddd956e83a2dfb09 (patch) | |
| tree | d139e1a770bcfc182f1aef38a1b908d634f9bb1c /src/core/cpu/kernels/CpuElementwiseUnaryKernel.h | |
| parent | 201e0fee596dafcf9c869a550fae29779aad2394 (diff) | |
| download | ComputeLibrary-668ccdcfb81bfab3a2d44cd1ddd956e83a2dfb09.tar.gz | |
Add dynamic tensor support to CpuElementwise
The kernels and operators for binary and unary operations
are now capable of being configured with dynamic shapes and
computing windows at run-time.
Additionally, changing arguments' names is done
for consistency.
Partially Implements: COMPMID-4127
Change-Id: I48e5038692db667dec7cb2b2906fe5683214fe19
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4973
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez Tello <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/cpu/kernels/CpuElementwiseUnaryKernel.h')
| -rw-r--r-- | src/core/cpu/kernels/CpuElementwiseUnaryKernel.h | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/src/core/cpu/kernels/CpuElementwiseUnaryKernel.h b/src/core/cpu/kernels/CpuElementwiseUnaryKernel.h index 193f6f1e4f..ceb90dcf70 100644 --- a/src/core/cpu/kernels/CpuElementwiseUnaryKernel.h +++ b/src/core/cpu/kernels/CpuElementwiseUnaryKernel.h @@ -38,7 +38,7 @@ namespace kernels /** Interface for an element-wise unary operation kernel * * Element-wise operation is computed by: - * @f[ output(x) = OP(input(x))@f] + * @f[ dst(x) = OP(src(x))@f] * */ class CpuElementwiseUnaryKernel : public ICpuKernel @@ -56,21 +56,21 @@ public: /** Function to configure the @ref CpuElementwiseUnaryKernel * - * @param[in] op Arithmetic operation to be executed. - * @param[in] input First tensor input. Data types supported: F16/F32, F16/F32/S32 for NEG/ABS operations. - * @param[out] output Output tensor. Data types supported: Same as @p input. + * @param[in] op Arithmetic operation to be executed. + * @param[in] src First tensor input. Data types supported: F16/F32, F16/F32/S32 for NEG/ABS operations. + * @param[out] dst Output tensor. Data types supported: Same as @p src. */ - void configure(ElementWiseUnary op, const ITensorInfo &input, ITensorInfo &output); + void configure(ElementWiseUnary op, const ITensorInfo &src, ITensorInfo &dst); /** Static function to check if given info will lead to a valid configuration of @ref CpuElementwiseUnaryKernel * - * @param[in] op Arithmetic operation to be executed. - * @param[in] input First tensor input info. Data types supported: F16/F32, F16/F32/S32 for NEG/ABS operations. - * @param[in] output Output tensor info. Data types supported: Same as @p input. + * @param[in] op Arithmetic operation to be executed. + * @param[in] src First tensor input info. Data types supported: F16/F32, F16/F32/S32 for NEG/ABS operations. + * @param[in] dst Output tensor info. Data types supported: Same as @p src. * * @return a Status */ - static Status validate(ElementWiseUnary op, const ITensorInfo &input, const ITensorInfo &output); + static Status validate(ElementWiseUnary op, const ITensorInfo &src, const ITensorInfo &dst); // Inherited methods overridden: void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override; |