From f1c2bf0971dd1c996da149faf3dd669d566074c7 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Wed, 13 Jun 2018 14:05:54 +0100 Subject: COMPMID-1201 - Implementing Winograd Convolution Layer 1x3 and 3x1 kernels on OpenCL Change-Id: I39667bab49daa4da009694163274a59fd3574c73 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/137595 Tested-by: Jenkins Reviewed-by: Giorgio Arena Reviewed-by: Georgios Pinitas --- arm_compute/core/CL/CLHelpers.h | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'arm_compute/core/CL/CLHelpers.h') diff --git a/arm_compute/core/CL/CLHelpers.h b/arm_compute/core/CL/CLHelpers.h index 1054f9a615..3b025cc5bb 100644 --- a/arm_compute/core/CL/CLHelpers.h +++ b/arm_compute/core/CL/CLHelpers.h @@ -109,5 +109,15 @@ bool arm_non_uniform_workgroup_supported(const cl::Device &device); * @return True if the extension is supported */ bool dot8_supported(const cl::Device &device); + +/** This function checks if the Winograd configuration (defined through the output tile, kernel size and the data layout) is supported on OpenCL + * + * @param[in] output_tile Output tile for the Winograd filtering algorithm + * @param[in] kernel_size Kernel size for the Winograd filtering algorithm + * @param[in] data_layout Data layout of the input tensor + * + * @return True if the configuration is supported + */ +bool cl_winograd_convolution_layer_supported(const Size2D &output_tile, const Size2D &kernel_size, DataLayout data_layout); } #endif /* __ARM_COMPUTE_CLHELPERS_H__ */ -- cgit v1.2.1