diff options
| author | Usama Arif <usama.arif@arm.com> | 2019-05-20 14:21:40 +0100 |
|---|---|---|
| committer | Usama Arif <usama.arif@arm.com> | 2019-05-22 11:58:36 +0000 |
| commit | ac33d7eb7bf0db65c3663d9707e509ca91337349 (patch) | |
| tree | 3f454d2632fad897f6559a9dffb0dd512492e3ec /arm_compute/runtime | |
| parent | 1f103d3bb2191d70806c9d9c0113aa5602079828 (diff) | |
| download | ComputeLibrary-ac33d7eb7bf0db65c3663d9707e509ca91337349.tar.gz | |
COMPMID-2264: Implement LOG operator for CL
Change-Id: I95ca659e8884191762b5d611a424a6e8a1d88885
Signed-off-by: Usama Arif <usama.arif@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1189
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime')
| -rw-r--r-- | arm_compute/runtime/CL/functions/CLElementWiseUnaryLayer.h | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/arm_compute/runtime/CL/functions/CLElementWiseUnaryLayer.h b/arm_compute/runtime/CL/functions/CLElementWiseUnaryLayer.h index b8a6fa8a81..de4bc59240 100644 --- a/arm_compute/runtime/CL/functions/CLElementWiseUnaryLayer.h +++ b/arm_compute/runtime/CL/functions/CLElementWiseUnaryLayer.h @@ -110,6 +110,26 @@ public: static Status validate(const ITensorInfo *input, const ITensorInfo *output); }; +/** Basic function to perform elementwise log on an input tensor. */ +class CLLogLayer : public ICLSimpleFunction +{ +public: + /** Initialize the function + * + * @param[in] input Input tensor. Data types supported: F16/F32. + * @param[out] output Output tensor. Data types supported: same as @p input. + */ + void configure(const ICLTensor *input, ICLTensor *output); + /** Static function to check if given info will lead to a valid configuration of @ref CLLogLayer + * + * @param[in] input First tensor input info. Data types supported: F16/F32. + * @param[in] output Output tensor info. Data types supported: Same as @p input. + * + * @return a status + */ + static Status validate(const ITensorInfo *input, const ITensorInfo *output); +}; + /** Basic function to get the absolute value of an input tensor. */ class CLAbsLayer : public ICLSimpleFunction { |