commit a341791df04b05fdeeaa34df485b9a577e3736cf
author Georgios Pinitas <georgios.pinitas@arm.com> Thu Jun 17 18:51:30 2021 +0100
committer Pablo Marquez Tello <pablo.tello@arm.com> Fri Jun 18 06:07:21 2021 +0000
tree 88de41cc7ed7edfec488cd475a0abf6d940e76d8
parent 08302c17cd57356b35d46e17dc8d8f76672da5cf

Remove implementation headers from NESoftmaxLayer public header

Resolves: COMPMID-4587

Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Ib216abcb0b9cd7f545d7c97e9d3447cb1b28f180
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5828
Reviewed-by: Pablo Marquez Tello <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

arm_compute/runtime/NEON/functions/NESoftmaxLayer.h

diff --git a/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h b/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h
index 02d0cc1..1787de6 100644
--- a/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h
+++ b/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h
@@ -24,8 +24,10 @@
 #ifndef ARM_COMPUTE_NESOFTMAXLAYER_H
 #define ARM_COMPUTE_NESOFTMAXLAYER_H
 
+#include "arm_compute/core/Error.h"
 #include "arm_compute/runtime/IFunction.h"
-#include "src/core/helpers/MemoryHelpers.h"
+#include "arm_compute/runtime/IMemoryManager.h"
+
 #include <memory>
 
 namespace arm_compute

src/runtime/NEON/functions/NESoftmaxLayer.cpp

diff --git a/src/runtime/NEON/functions/NESoftmaxLayer.cpp b/src/runtime/NEON/functions/NESoftmaxLayer.cpp
index af8546d..bee692c 100644
--- a/src/runtime/NEON/functions/NESoftmaxLayer.cpp
+++ b/src/runtime/NEON/functions/NESoftmaxLayer.cpp
@@ -26,6 +26,7 @@
 #include "arm_compute/runtime/MemoryGroup.h"
 #include "arm_compute/runtime/Tensor.h"
 #include "src/core/cpu/kernels/CpuSoftmaxKernel.h"
+#include "src/core/helpers/MemoryHelpers.h"
 #include "src/core/helpers/SoftmaxHelpers.h"
 #include "src/runtime/cpu/operators/CpuSoftmax.h"