diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2021-04-16 15:08:59 +0100 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2021-06-24 11:16:30 +0000 |
| commit | 561c176598cd14245e2e7918fdf136d1c888d1da (patch) | |
| tree | 82adfff6de30292dabbbcc7ced4ae35cac3d45cf /arm_compute/core/CL | |
| parent | 31c7c26822270f1c4952c8973aa8bfb38e0a7c68 (diff) | |
| download | ComputeLibrary-561c176598cd14245e2e7918fdf136d1c888d1da.tar.gz | |
Rework OpenCL Depthwise Convolution
- Remove dedicated kernels for NCHW. Now we only use NHWC with permute
- Remove specialized kernels for 3x3 NHWC
- Simplify CLDepthwiseConvolutionLayer.cpp to call just the native
implementation for both floating-point and quantized data types
- Develop two parametric opencl kernels for depthwise convolution layer NHWC
(floating-point and quantized)
- Add support to export the weights to cl_image
- Extend test for depthwise convolution on opencl
Resolves COMPMID-4417
Change-Id: I253dd5d959a70783c82e62b1771a5e9f91621cb0
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5806
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Diffstat (limited to 'arm_compute/core/CL')
| -rw-r--r-- | arm_compute/core/CL/CLHelpers.h | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/arm_compute/core/CL/CLHelpers.h b/arm_compute/core/CL/CLHelpers.h index a9ac6a5933..180211c558 100644 --- a/arm_compute/core/CL/CLHelpers.h +++ b/arm_compute/core/CL/CLHelpers.h @@ -236,5 +236,13 @@ bool get_wbsm_support_info(const cl::Device &device); */ void set_wbsm(cl::Kernel &kernel, cl_int wbsm_hint); +/* Helper function to check if we can export the weights to cl_image + * + * @param[in] tensor Weights tensor + * + * @return true if we can export the weights to cl_image + */ +bool export_weights_to_cl_image(const ITensorInfo *tensor); + } // namespace arm_compute #endif /* ARM_COMPUTE_CLHELPERS_H */ |