diff options
| author | Isabella Gottardi <isabella.gottardi@arm.com> | 2018-12-14 11:40:40 +0000 |
|---|---|---|
| committer | Isabella Gottardi <isabella.gottardi@arm.com> | 2018-12-17 14:02:06 +0000 |
| commit | cc6129c06af98616a0e4d68475cfa3d92aaf63b3 (patch) | |
| tree | b44e5804492d45544a0f4034dfac35ddb86c528f /arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h | |
| parent | 49b1015f3505b920f9e4495017c99f91eed68965 (diff) | |
| download | ComputeLibrary-cc6129c06af98616a0e4d68475cfa3d92aaf63b3.tar.gz | |
COMPMID-1812: CLSpaceToBatch paddings not calculated correctly
Change-Id: I63fed6799c4ed2848ff80cd7458124692a52bb98
Reviewed-on: https://review.mlplatform.org/400
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h')
| -rw-r--r-- | arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h b/arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h index 6478774701..9b1dd979ec 100644 --- a/arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h +++ b/arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h @@ -26,6 +26,7 @@ #include "arm_compute/runtime/IFunction.h" +#include "arm_compute/core/CL/kernels/CLMemsetKernel.h" #include "arm_compute/core/CL/kernels/CLSpaceToBatchLayerKernel.h" #include "arm_compute/core/Types.h" #include "arm_compute/runtime/CL/CLTensor.h" @@ -34,7 +35,11 @@ namespace arm_compute { class ICLTensor; -/** Basic function to run @ref CLSpaceToBatchLayerKernel. */ +/** Basic function to spatial divide a tensor. This function calls the following OpenCL kernels/functions: + * + * -# @ref CLMemsetKernel + * -# @ref CLSpaceToBatchLayerKernel + */ class CLSpaceToBatchLayer : public IFunction { public: @@ -96,7 +101,7 @@ public: private: CLSpaceToBatchLayerKernel _space_to_batch_kernel; /**< SpaceToBatch kernel to run */ - ICLTensor *_output; /**< Output tensor */ + CLMemsetKernel _memset_kernel; /**< Memset kernel to run */ bool _has_padding; /**< Flag to check if the output has padding */ }; } // namespace arm_compute |