aboutsummaryrefslogtreecommitdiff
path: root/src/graph/mutators/InPlaceOperationMutator.cpp
diff options
context:
space:
mode:
authorManuel Bottini <manuel.bottini@arm.com>2020-06-03 13:20:41 +0100
committerManuel Bottini <manuel.bottini@arm.com>2020-06-22 16:23:58 +0000
commit80feed5193de6b10d8ab65b42fb988c241c5d09d (patch)
treed7b695d0c3d099e7bbdbd10a82fd355d77f0fdfc /src/graph/mutators/InPlaceOperationMutator.cpp
parent0028d7c7230d3fda419db5c6d5d8141830bd13f9 (diff)
downloadComputeLibrary-80feed5193de6b10d8ab65b42fb988c241c5d09d.tar.gz
COMPMID-3479: Perform in-place computations in NEElementwiseUnaryKernel
Change-Id: I2102bfe95c2c2335bb587842f9d860cf939a9026 Signed-off-by: Manuel Bottini <manuel.bottini@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3315 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/graph/mutators/InPlaceOperationMutator.cpp')
-rw-r--r--src/graph/mutators/InPlaceOperationMutator.cpp1
1 files changed, 1 insertions, 0 deletions
diff --git a/src/graph/mutators/InPlaceOperationMutator.cpp b/src/graph/mutators/InPlaceOperationMutator.cpp
index ba80d8a26a..394dba84ff 100644
--- a/src/graph/mutators/InPlaceOperationMutator.cpp
+++ b/src/graph/mutators/InPlaceOperationMutator.cpp
@@ -47,6 +47,7 @@ void InPlaceOperationMutator::mutate(Graph &g)
NodeType::ActivationLayer,
NodeType::BatchNormalizationLayer,
NodeType::EltwiseLayer,
+ NodeType::UnaryEltwiseLayer,
NodeType::PrintLayer
};