diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-12-03 20:37:43 +0000 |
|---|---|---|
| committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-12-08 11:56:12 +0000 |
| commit | e2696b1f9bb28b69beff99f54addd48f60823ddb (patch) | |
| tree | 68705f1cdff45e4d0c174b6037f6e5ff696717d2 /arm_compute/runtime/CL/functions/CLFlattenLayer.h | |
| parent | 8c3c0e7b117723bc98b6acc85565ffa521b10c0d (diff) | |
| download | ComputeLibrary-e2696b1f9bb28b69beff99f54addd48f60823ddb.tar.gz | |
Wrap Flatten layer over reshape
Flatten layer is lowered into a Reshape layer.
Remove (CL/NE)FlatternLayerKernel.
Partially Resolves: COMPMID-3996
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Id9e2ddfe2e2dd793541badff3490c05e4c908f88
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4660
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLFlattenLayer.h')
| -rw-r--r-- | arm_compute/runtime/CL/functions/CLFlattenLayer.h | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/arm_compute/runtime/CL/functions/CLFlattenLayer.h b/arm_compute/runtime/CL/functions/CLFlattenLayer.h index f5f4ff554f..ffe06aa610 100644 --- a/arm_compute/runtime/CL/functions/CLFlattenLayer.h +++ b/arm_compute/runtime/CL/functions/CLFlattenLayer.h @@ -25,7 +25,8 @@ #define ARM_COMPUTE_CLFLATTENLAYER_H #include "arm_compute/core/Types.h" -#include "arm_compute/runtime/CL/ICLSimpleFunction.h" +#include "arm_compute/runtime/CL/functions/CLReshapeLayer.h" +#include "arm_compute/runtime/IFunction.h" namespace arm_compute { @@ -38,7 +39,7 @@ class ITensorInfo; * -# @ref CLFlattenLayerKernel * */ -class CLFlattenLayer : public ICLSimpleFunction +class CLFlattenLayer : public IFunction { public: /** Initialise the kernel's input and output. @@ -68,6 +69,12 @@ public: * @return a status */ static Status validate(const ITensorInfo *input, const ITensorInfo *output); + + // Inherited methods overridden: + void run() override; + +private: + CLReshapeLayer _reshape{}; }; } // namespace arm_compute |