diff options
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLHarrisCorners.h')
| -rw-r--r-- | arm_compute/runtime/CL/functions/CLHarrisCorners.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/arm_compute/runtime/CL/functions/CLHarrisCorners.h b/arm_compute/runtime/CL/functions/CLHarrisCorners.h index f9a1275f68..e09e67060f 100644 --- a/arm_compute/runtime/CL/functions/CLHarrisCorners.h +++ b/arm_compute/runtime/CL/functions/CLHarrisCorners.h @@ -31,11 +31,12 @@ #include "arm_compute/core/CL/kernels/CLHarrisCornersKernel.h" #include "arm_compute/core/NEON/kernels/NEHarrisCornersKernel.h" #include "arm_compute/core/Types.h" +#include "arm_compute/runtime/CL/CLMemoryGroup.h" #include "arm_compute/runtime/CL/CLTensor.h" #include "arm_compute/runtime/CL/functions/CLNonMaximaSuppression3x3.h" +#include "arm_compute/runtime/IMemoryManager.h" #include <cstdint> - #include <memory> namespace arm_compute @@ -60,7 +61,7 @@ class CLHarrisCorners : public IFunction { public: /** Constructor */ - CLHarrisCorners(); + CLHarrisCorners(std::shared_ptr<IMemoryManager> memory_manager = nullptr); /** Prevent instances of this class from being copied (As this class contains pointers) */ CLHarrisCorners(const CLHarrisCorners &) = delete; /** Prevent instances of this class from being copied (As this class contains pointers) */ @@ -85,6 +86,7 @@ public: void run() override; private: + CLMemoryGroup _memory_group; /**< Function's memory group */ std::unique_ptr<IFunction> _sobel; /**< Sobel function */ CLHarrisScoreKernel _harris_score; /**< Harris score kernel */ CLNonMaximaSuppression3x3 _non_max_suppr; /**< Non-maxima suppression function */ |