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>
diff --git a/arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h b/arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h
index 1367f37..5871cc5 100644
--- a/arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h
+++ b/arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h
@@ -82,8 +82,24 @@
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;