aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/CL/functions/CLFlattenLayer.h
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2021-05-17 02:42:04 +0100
committerGeorgios Pinitas <georgios.pinitas@arm.com>2021-05-18 09:27:51 +0000
commit494b97a8297c18d0d610a54189c8f848e2cf5446 (patch)
treeda72ac4fd0763780aba341b88f9e40ec9510ccf8 /arm_compute/runtime/CL/functions/CLFlattenLayer.h
parent9e1f074af3a71b5bd5bd87b292c74fb57a90347c (diff)
downloadComputeLibrary-494b97a8297c18d0d610a54189c8f848e2cf5446.tar.gz
Port CLFlattenLayer to a memory injecting interface
Resolves: COMPMID-4502 Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com> Change-Id: I00621a772135f65a92aa70e6a19dd5f743915378 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5649 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLFlattenLayer.h')
-rw-r--r--arm_compute/runtime/CL/functions/CLFlattenLayer.h22
1 files changed, 15 insertions, 7 deletions
diff --git a/arm_compute/runtime/CL/functions/CLFlattenLayer.h b/arm_compute/runtime/CL/functions/CLFlattenLayer.h
index d2c37b1c22..182e97dc67 100644
--- a/arm_compute/runtime/CL/functions/CLFlattenLayer.h
+++ b/arm_compute/runtime/CL/functions/CLFlattenLayer.h
@@ -25,23 +25,30 @@
#define ARM_COMPUTE_CLFLATTENLAYER_H
#include "arm_compute/core/Types.h"
-#include "arm_compute/runtime/CL/functions/CLReshapeLayer.h"
#include "arm_compute/runtime/IFunction.h"
+#include <memory>
namespace arm_compute
{
class CLCompileContext;
class ICLTensor;
class ITensorInfo;
-/** Basic function to execute flatten. This function calls the following OpenCL kernel:
-*
-* -# @ref CLReshapeLayer
-*
-*/
+/** Basic function to execute flatten */
class CLFlattenLayer : public IFunction
{
public:
+ CLFlattenLayer();
+ /** Destructor */
+ ~CLFlattenLayer();
+ /** Prevent instances of this class from being copied (As this class contains pointers) */
+ CLFlattenLayer(const CLFlattenLayer &) = delete;
+ /** Default move constructor */
+ CLFlattenLayer(CLFlattenLayer &&);
+ /** Prevent instances of this class from being copied (As this class contains pointers) */
+ CLFlattenLayer &operator=(const CLFlattenLayer &) = delete;
+ /** Default move assignment operator */
+ CLFlattenLayer &operator=(CLFlattenLayer &&);
/** Initialise the kernel's input and output.
*
* Valid data layouts:
@@ -82,7 +89,8 @@ public:
void run() override;
private:
- CLReshapeLayer _reshape{};
+ struct Impl;
+ std::unique_ptr<Impl> _impl;
};
} // namespace arm_compute