From f3c52596a87eb4ccada601fc57f612bab137415d Mon Sep 17 00:00:00 2001 From: Manuel Bottini Date: Wed, 12 May 2021 15:39:07 +0100 Subject: Remove padding from NERemapKernel Use of out_of_tensor function to check if parallel instructons can be used safely Reverting to serial computation otherwise Resolves: COMPMID-4449 Change-Id: I23a986612e3c5d0367e23e56f1aeedbb1330cffc Signed-off-by: Manuel Bottini Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5651 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Georgios Pinitas --- arm_compute/runtime/NEON/functions/NERemap.h | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'arm_compute') diff --git a/arm_compute/runtime/NEON/functions/NERemap.h b/arm_compute/runtime/NEON/functions/NERemap.h index 1693078f66..271ac9739b 100644 --- a/arm_compute/runtime/NEON/functions/NERemap.h +++ b/arm_compute/runtime/NEON/functions/NERemap.h @@ -25,7 +25,7 @@ #define ARM_COMPUTE_NEREMAP_H #include "arm_compute/core/Types.h" -#include "arm_compute/runtime/NEON/INESimpleFunction.h" +#include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h" #include "arm_compute/runtime/Tensor.h" #include @@ -36,10 +36,9 @@ class ITensor; /** Basic function to execute remap. This function calls the following kernels: * - * -# @ref NEFillBorderKernel (executed if border_mode == CONSTANT or border_mode == REPLICATE) * -# @ref NERemapKernel */ -class NERemap : public INESimpleFunction +class NERemap : public INESimpleFunctionNoBorder { public: /** Initialise the function's sources, destination, interpolation policy and border mode. @@ -58,7 +57,7 @@ public: * @param[out] output Output tensor. Data type supported: U8. * @param[in] policy Interpolation policy to use. Only NEAREST and BILINEAR are supported. * @param[in] border_mode Border mode to use on the input tensor. - * @param[in] constant_border_value (Optional) Constant value to use for borders if border_mode is set to CONSTANT. + * @param[in] constant_border_value (Optional) Constant value to use for borders if border_mode is set to CONSTANT. Defaults to 0. * */ void configure(ITensor *input, const ITensor *map_x, const ITensor *map_y, ITensor *output, -- cgit v1.2.1