aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/functions/NEElementwiseUnaryLayer.cpp
diff options
context:
space:
mode:
authorUsama Arif <usama.arif@arm.com>2019-05-10 12:06:28 +0100
committerUsama Arif <usama.arif@arm.com>2019-05-13 09:01:24 +0000
commitf6e475c9a092bc6e0fb53f484fbf2832183a9c44 (patch)
treea6a5f1cd6b48a8615fd942639d2b52c61d5c46be /src/runtime/NEON/functions/NEElementwiseUnaryLayer.cpp
parentbcfd09a14a61bb8457555c61e7c5697b1f901ddb (diff)
downloadComputeLibrary-f6e475c9a092bc6e0fb53f484fbf2832183a9c44.tar.gz
COMPMID-2268: Implement NEG for NEON.
Change-Id: I90c023dbea8ea12e9af677294ba576b2bfcc02a4 Signed-off-by: Usama Arif <usama.arif@arm.com> Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/184216 Tested-by: bsgcomp <bsgcomp@arm.com> Comments-Addressed: bsgcomp <bsgcomp@arm.com> Reviewed-by: Pablo Tello <pablo.tello@arm.com> Reviewed-on: https://review.mlplatform.org/c/1099 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Pablo Marquez <pablo.tello@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEElementwiseUnaryLayer.cpp')
-rw-r--r--src/runtime/NEON/functions/NEElementwiseUnaryLayer.cpp12
1 files changed, 12 insertions, 0 deletions
diff --git a/src/runtime/NEON/functions/NEElementwiseUnaryLayer.cpp b/src/runtime/NEON/functions/NEElementwiseUnaryLayer.cpp
index 10142c732f..48f4975b1a 100644
--- a/src/runtime/NEON/functions/NEElementwiseUnaryLayer.cpp
+++ b/src/runtime/NEON/functions/NEElementwiseUnaryLayer.cpp
@@ -51,4 +51,16 @@ Status NEExpLayer::validate(const ITensorInfo *input, const ITensorInfo *output)
{
return NEElementwiseUnaryKernel::validate(ElementWiseUnary::EXP, input, output);
}
+
+void NENegLayer::configure(const ITensor *input, ITensor *output)
+{
+ auto k = arm_compute::support::cpp14::make_unique<NEElementwiseUnaryKernel>();
+ k->configure(ElementWiseUnary::NEG, input, output);
+ _kernel = std::move(k);
+}
+Status NENegLayer::validate(const ITensorInfo *input, const ITensorInfo *output)
+{
+ return NEElementwiseUnaryKernel::validate(ElementWiseUnary::NEG, input, output);
+}
+
} // namespace arm_compute