diff options
Diffstat (limited to 'src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp')
-rw-r--r-- | src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp b/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp index be3c7e25da..e804ef1853 100644 --- a/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp +++ b/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp @@ -84,6 +84,9 @@ void CLElementWiseUnaryLayerKernel::configure(const ICLTensor *input, ICLTensor case ElementWiseUnary::EXP: build_opts.add_option("-DOPERATION=exponential"); break; + case ElementWiseUnary::NEG: + build_opts.add_option("-DOPERATION=neg"); + break; default: ARM_COMPUTE_ERROR("Not implemented"); } @@ -116,4 +119,4 @@ void CLElementWiseUnaryLayerKernel::run(const Window &window, cl::CommandQueue & enqueue(queue, *this, collapsed); } while(window.slide_window_slice_1D(collapsed)); -}
\ No newline at end of file +} |