From cc6129c06af98616a0e4d68475cfa3d92aaf63b3 Mon Sep 17 00:00:00 2001 From: Isabella Gottardi Date: Fri, 14 Dec 2018 11:40:40 +0000 Subject: COMPMID-1812: CLSpaceToBatch paddings not calculated correctly Change-Id: I63fed6799c4ed2848ff80cd7458124692a52bb98 Reviewed-on: https://review.mlplatform.org/400 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Michalis Spyrou --- arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h') 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 -- cgit v1.2.1