diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-06-13 14:05:54 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:53:57 +0000 |
| commit | f1c2bf0971dd1c996da149faf3dd669d566074c7 (patch) | |
| tree | 802b3ce5198c3209d77fc6b603c209023fe45650 /arm_compute/core/CL/CLHelpers.h | |
| parent | 89a2b571cfc0ea87c26ba8b1ed1ab87d13244f0e (diff) | |
| download | ComputeLibrary-f1c2bf0971dd1c996da149faf3dd669d566074c7.tar.gz | |
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 <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'arm_compute/core/CL/CLHelpers.h')
| -rw-r--r-- | arm_compute/core/CL/CLHelpers.h | 10 |
1 files changed, 10 insertions, 0 deletions
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__ */ |