diff options
Diffstat (limited to 'arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h')
-rw-r--r-- | arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h | 20 |
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; |