COMPMID-1005: Update Depthwise Convolution form RSH

Change-Id: I3033ddb8de183661010d6c71a83f71132037b139
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/124338
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>

1 files changed, 18 insertions, 2 deletions

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<depthwise::IDepthwiseConvolution> 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<depthwise::IDepthwiseConvolution> create_convolver_object(PadStrideInfo  conv_info,
+                                                                              const ITensor *w,
+                                                                              const ITensor *in,
+                                                                              ITensor       *out,
+                                                                              bool           setup_strides = false);
 
 private:
     BorderSize                                        _border_size;