From f24411ffc842970609a1fb6ba2f9527cfb681dbd Mon Sep 17 00:00:00 2001 From: giuros01 Date: Thu, 16 May 2019 11:47:13 +0100 Subject: COMPMID-2255: Implement ABS operator for CL Change-Id: I8867953f8ae84a77f126aedd32b2f98c062905b1 Signed-off-by: giuros01 Reviewed-on: https://review.mlplatform.org/c/1151 Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Reviewed-by: Georgios Pinitas --- src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) (limited to 'src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp') diff --git a/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp b/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp index b6484ec737..15823b06ce 100644 --- a/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp +++ b/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp @@ -79,16 +79,19 @@ void CLElementWiseUnaryLayerKernel::configure(const ICLTensor *input, ICLTensor switch(op) { case ElementWiseUnary::RSQRT: - build_opts.add_option("-DOPERATION=inverse_sqrt"); + build_opts.add_option("-DOPERATION=rsqrt_op"); break; case ElementWiseUnary::EXP: - build_opts.add_option("-DOPERATION=exponential"); + build_opts.add_option("-DOPERATION=exp_op"); break; case ElementWiseUnary::NEG: - build_opts.add_option("-DOPERATION=neg"); + build_opts.add_option("-DOPERATION=neg_op"); break; case ElementWiseUnary::SIN: - build_opts.add_option("-DOPERATION=sine"); + build_opts.add_option("-DOPERATION=sin_op"); + break; + case ElementWiseUnary::ABS: + build_opts.add_option("-DOPERATION=fabs_op"); break; default: ARM_COMPUTE_ERROR("Not implemented"); -- cgit v1.2.1