diff options
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/NEON/kernels/NEElementwiseUnaryKernel.h | 8 | ||||
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEElementwiseUnaryLayer.h | 18 |
2 files changed, 13 insertions, 13 deletions
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 |