diff options
author | Usama Arif <usama.arif@arm.com> | 2019-05-24 14:53:59 +0100 |
---|---|---|
committer | Usama Arif <usama.arif@arm.com> | 2019-05-24 16:36:10 +0000 |
commit | 6a4d5426637b6cc967665e0455609535938c211d (patch) | |
tree | 4bcd6a9c4b05f0f979c46074f309f27a3355f907 /src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp | |
parent | 0a5a57a3f794de851408bae1c63b1660b4c5cbe7 (diff) | |
download | ComputeLibrary-6a4d5426637b6cc967665e0455609535938c211d.tar.gz |
COMPMID-2159 Implement Round for CL
Change-Id: I9d3ed4bb3cf7bef39bd0e71b3c6f266ff8187290
Signed-off-by: Usama Arif <usama.arif@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1226
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp')
-rw-r--r-- | src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp b/src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp index b1e18ecd7e..b9ebf6972e 100644 --- a/src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp +++ b/src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp @@ -95,4 +95,15 @@ Status CLLogLayer::validate(const ITensorInfo *input, const ITensorInfo *output) return CLElementWiseUnaryLayerKernel::validate(input, output, ElementWiseUnary::LOG); } +void CLRoundLayer::configure(const ICLTensor *input, ICLTensor *output) +{ + auto k = arm_compute::support::cpp14::make_unique<CLElementWiseUnaryLayerKernel>(); + k->configure(input, output, ElementWiseUnary::ROUND); + _kernel = std::move(k); +} +Status CLRoundLayer::validate(const ITensorInfo *input, const ITensorInfo *output) +{ + return CLElementWiseUnaryLayerKernel::validate(input, output, ElementWiseUnary::ROUND); +} + } // namespace arm_compute |