From 80feed5193de6b10d8ab65b42fb988c241c5d09d Mon Sep 17 00:00:00 2001 From: Manuel Bottini Date: Wed, 3 Jun 2020 13:20:41 +0100 Subject: COMPMID-3479: Perform in-place computations in NEElementwiseUnaryKernel Change-Id: I2102bfe95c2c2335bb587842f9d860cf939a9026 Signed-off-by: Manuel Bottini Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3315 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Georgios Pinitas Comments-Addressed: Arm Jenkins --- .../core/NEON/kernels/NEElementwiseUnaryKernel.h | 8 ++++---- .../runtime/NEON/functions/NEElementwiseUnaryLayer.h | 18 +++++++++--------- 2 files changed, 13 insertions(+), 13 deletions(-) (limited to 'arm_compute') diff --git a/arm_compute/core/NEON/kernels/NEElementwiseUnaryKernel.h b/arm_compute/core/NEON/kernels/NEElementwiseUnaryKernel.h index 9a41cecf19..02c390b6ba 100644 --- a/arm_compute/core/NEON/kernels/NEElementwiseUnaryKernel.h +++ b/arm_compute/core/NEON/kernels/NEElementwiseUnaryKernel.h @@ -57,11 +57,11 @@ public: /** Default destructor */ ~NEElementwiseUnaryKernel() = default; - /** Static function to check if given info will lead to a valid configuration of @ref NEElementwiseUnaryKernel + /** Function to configure the @ref NEElementwiseUnaryKernel * - * @param[in] op Arithmetic operation to be executed. - * @param[in] input First tensor input. Data types supported: F16/F32. - * @param[in] output Output tensor. Data types supported: Same as @p input. + * @param[in] op Arithmetic operation to be executed. + * @param[in] input First tensor input. Data types supported: F16/F32. + * @param[out] output Output tensor. Data types supported: Same as @p input. */ void configure(ElementWiseUnary op, const ITensor *input, ITensor *output); diff --git a/arm_compute/runtime/NEON/functions/NEElementwiseUnaryLayer.h b/arm_compute/runtime/NEON/functions/NEElementwiseUnaryLayer.h index 094f875b35..1fd24887a5 100644 --- a/arm_compute/runtime/NEON/functions/NEElementwiseUnaryLayer.h +++ b/arm_compute/runtime/NEON/functions/NEElementwiseUnaryLayer.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2019 ARM Limited. + * Copyright (c) 2018-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -24,14 +24,14 @@ #ifndef ARM_COMPUTE_NEELEMENTWISEUNARYLAYER_H #define ARM_COMPUTE_NEELEMENTWISEUNARYLAYER_H -#include "arm_compute/runtime/NEON/INESimpleFunction.h" +#include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h" namespace arm_compute { class ITensor; /** Basic function to perform inverse square root on an input tensor. */ -class NERsqrtLayer : public INESimpleFunction +class NERsqrtLayer : public INESimpleFunctionNoBorder { public: /** Initialize the function @@ -51,7 +51,7 @@ public: }; /** Basic function to perform exponential on an input tensor. */ -class NEExpLayer : public INESimpleFunction +class NEExpLayer : public INESimpleFunctionNoBorder { public: /** Initialize the function @@ -71,7 +71,7 @@ public: }; /** Basic function to negate an input tensor. */ -class NENegLayer : public INESimpleFunction +class NENegLayer : public INESimpleFunctionNoBorder { public: /** Initialize the function @@ -91,7 +91,7 @@ public: }; /** Basic function to compute the natural logarithm of an input tensor. */ -class NELogLayer : public INESimpleFunction +class NELogLayer : public INESimpleFunctionNoBorder { public: /** Initialize the function @@ -111,7 +111,7 @@ public: }; /** Basic function to compute the absolute value of an input tensor. */ -class NEAbsLayer : public INESimpleFunction +class NEAbsLayer : public INESimpleFunctionNoBorder { public: /** Initialize the function @@ -131,7 +131,7 @@ public: }; /** Basic function to compute the round value elementwise of an input tensor. */ -class NERoundLayer : public INESimpleFunction +class NERoundLayer : public INESimpleFunctionNoBorder { public: /** Initialize the function @@ -151,7 +151,7 @@ public: }; /** Basic function to compute the sine of an input tensor. */ -class NESinLayer : public INESimpleFunction +class NESinLayer : public INESimpleFunctionNoBorder { public: /** Initialize the function -- cgit v1.2.1