aboutsummaryrefslogtreecommitdiff
path: root/arm_compute
diff options
context:
space:
mode:
authorManuel Bottini <manuel.bottini@arm.com>2021-06-09 16:37:32 +0100
committerManuel Bottini <manuel.bottini@arm.com>2021-06-15 16:31:27 +0000
commit94f799e8f6f605333d40472860fb472e8ba6d83d (patch)
treece528244814463ed42dc86a84d54ea870c75d592 /arm_compute
parent36dff9f81e3a95aea19fcc7246a4896930a14bc6 (diff)
downloadComputeLibrary-94f799e8f6f605333d40472860fb472e8ba6d83d.tar.gz
Fix incorrect memory handling in ported functions
Details of the functions: - ClSoftmax - CpuSoftmax - CpuPool2d Change-Id: Icd2c14d5df010c3b2301e2693ce6f414d7c61916 Resolves: COMPMID-4404 Signed-off-by: Manuel Bottini <manuel.bottini@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5797 Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute')
-rw-r--r--arm_compute/runtime/CL/functions/CLSoftmaxLayer.h3
-rw-r--r--arm_compute/runtime/NEON/functions/NEPoolingLayer.h2
-rw-r--r--arm_compute/runtime/NEON/functions/NESoftmaxLayer.h3
3 files changed, 1 insertions, 7 deletions
diff --git a/arm_compute/runtime/CL/functions/CLSoftmaxLayer.h b/arm_compute/runtime/CL/functions/CLSoftmaxLayer.h
index 721a47144e..687f8ff6d8 100644
--- a/arm_compute/runtime/CL/functions/CLSoftmaxLayer.h
+++ b/arm_compute/runtime/CL/functions/CLSoftmaxLayer.h
@@ -106,9 +106,6 @@ public:
private:
struct Impl;
std::unique_ptr<Impl> _impl;
-
- /** Allocate workspace required by the operator */
- void allocate_workspace();
};
using CLSoftmaxLayer = CLSoftmaxLayerGeneric<false>;
diff --git a/arm_compute/runtime/NEON/functions/NEPoolingLayer.h b/arm_compute/runtime/NEON/functions/NEPoolingLayer.h
index b5366fa1c1..9398e1fce9 100644
--- a/arm_compute/runtime/NEON/functions/NEPoolingLayer.h
+++ b/arm_compute/runtime/NEON/functions/NEPoolingLayer.h
@@ -95,8 +95,6 @@ public:
void run() override;
private:
- MemoryGroup _memory_group;
-
struct Impl;
std::unique_ptr<Impl> _impl;
};
diff --git a/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h b/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h
index efe959f14e..02d0cc15b2 100644
--- a/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h
+++ b/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h
@@ -25,7 +25,7 @@
#define ARM_COMPUTE_NESOFTMAXLAYER_H
#include "arm_compute/runtime/IFunction.h"
-#include "arm_compute/runtime/MemoryGroup.h"
+#include "src/core/helpers/MemoryHelpers.h"
#include <memory>
namespace arm_compute
@@ -88,7 +88,6 @@ public:
void run() override;
private:
- MemoryGroup _memory_group;
struct Impl;
std::unique_ptr<Impl> _impl;
};