From be0ae93c50bfa3e588111585025278daa8cb0694 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Tue, 13 Mar 2018 13:08:12 +0000 Subject: COMPMID-1005: Update Depthwise Convolution form RSH Change-Id: I3033ddb8de183661010d6c71a83f71132037b139 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/124338 Tested-by: Jenkins Reviewed-by: Pablo Tello --- .../kernels/NEDepthwiseConvolutionLayer3x3Kernel.h | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) (limited to 'arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h') diff --git a/arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h b/arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h index 1367f378f7..5871cc5dcb 100644 --- a/arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h +++ b/arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h @@ -82,8 +82,24 @@ private: void configure_optimized(); void run_generic(const Window &window, const ThreadInfo &info); void run_optimized(const Window &window, const ThreadInfo &info); - std::unique_ptr create_convolver_object(TensorShape shape, PadStrideInfo conv_info, - const uint8_t *w_ptr, uint8_t *in_ptr, uint8_t *out_ptr); + /** Creates an optimized backend convolver object + * + * @note Convolver of strides 1,2 and convolution size of 3 is currently supported + * + * @param[in] conv_info Padding and stride information to use for the convolution + * @param[in] w Weights tensor + * @param[in] in Input tensor + * @param[in] out Output tensor + * @param[in] setup_strides (Optional) Boolean to enable setting the strides of the tensors + * in the convolver in case of padding. Defaults to false + * + * @return A convolver object or nullptr if the configuration is not supported + */ + std::unique_ptr create_convolver_object(PadStrideInfo conv_info, + const ITensor *w, + const ITensor *in, + ITensor *out, + bool setup_strides = false); private: BorderSize _border_size; -- cgit v1.2.1