diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-03-13 13:08:12 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:16 +0000 |
commit | be0ae93c50bfa3e588111585025278daa8cb0694 (patch) | |
tree | d78c13e8846c31587a5acb70b38b13fa7d03200d /arm_compute/core/NEON/kernels/NEDepthwiseConvolutionLayer3x3Kernel.h | |
parent | ae4ce7b411d0f4809ac7d3d90fe89bdb2520dbf6 (diff) | |
download | ComputeLibrary-be0ae93c50bfa3e588111585025278daa8cb0694.tar.gz |
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>
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; |