diff options
Diffstat (limited to 'src/core/CL')
-rw-r--r-- | src/core/CL/cl_kernels/elementwise_unary.cl | 2 | ||||
-rw-r--r-- | src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp | 3 |
2 files changed, 5 insertions, 0 deletions
diff --git a/src/core/CL/cl_kernels/elementwise_unary.cl b/src/core/CL/cl_kernels/elementwise_unary.cl index 2650654e4a..14d6f5dbe1 100644 --- a/src/core/CL/cl_kernels/elementwise_unary.cl +++ b/src/core/CL/cl_kernels/elementwise_unary.cl @@ -32,6 +32,8 @@ #define inverse_sqrt(input) rsqrt(input) // Calculate negative #define neg(input) (-input) +// Calculate sine +#define sine(input) sin(input) /** Applies element wise unary operator in a tensor. * diff --git a/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp b/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp index e804ef1853..b6484ec737 100644 --- a/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp +++ b/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp @@ -87,6 +87,9 @@ void CLElementWiseUnaryLayerKernel::configure(const ICLTensor *input, ICLTensor case ElementWiseUnary::NEG: build_opts.add_option("-DOPERATION=neg"); break; + case ElementWiseUnary::SIN: + build_opts.add_option("-DOPERATION=sine"); + break; default: ARM_COMPUTE_ERROR("Not implemented"); } |