diff options
author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2021-04-06 10:11:58 +0100 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-04-06 11:50:20 +0000 |
commit | 8efed67f607a0b733bd1794956559e5bb11db28c (patch) | |
tree | de12141302f638d8eb4bb45070e65c19f8e89694 | |
parent | ca62c6f53eb7244e6fed9f7e932608aa2496d9eb (diff) | |
download | ComputeLibrary-8efed67f607a0b733bd1794956559e5bb11db28c.tar.gz |
Fix build error in CLROIPoolingLayer
- Adds include for Status class definition
- Adds forward declaration of ITensorInfo
Resolves: COMPMID-4351
Change-Id: I8683aadb6f29ce12730ebfae67f1fe5cc89824ec
Signed-off-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5359
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
-rw-r--r-- | arm_compute/runtime/CL/functions/CLROIPoolingLayer.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/arm_compute/runtime/CL/functions/CLROIPoolingLayer.h b/arm_compute/runtime/CL/functions/CLROIPoolingLayer.h index 3b7804c2b7..a4c5c76f2e 100644 --- a/arm_compute/runtime/CL/functions/CLROIPoolingLayer.h +++ b/arm_compute/runtime/CL/functions/CLROIPoolingLayer.h @@ -24,6 +24,7 @@ #ifndef ARM_COMPUTE_CLROIPOOLINGLAYER_H #define ARM_COMPUTE_CLROIPOOLINGLAYER_H +#include "arm_compute/core/Error.h" #include "arm_compute/runtime/CL/ICLSimpleFunction.h" namespace arm_compute @@ -31,6 +32,7 @@ namespace arm_compute class CLCompileContext; class ICLTensor; class ROIPoolingLayerInfo; +class ITensorInfo; /** Basic function to run @ref CLROIPoolingLayerKernel. * @@ -87,5 +89,5 @@ public: */ static Status validate(const ITensorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info); }; -} +} // namespace arm_compute #endif /* ARM_COMPUTE_CLROIPOOLINGLAYER_H */ |