diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2022-05-30 14:41:49 +0100 |
---|---|---|
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2022-07-08 13:48:51 +0000 |
commit | 2cc50b39613cea5e55c8a4851ee08d284a3d4f66 (patch) | |
tree | 7faac6aaa4409bb127a9bd8ebc89056dae40066e /arm_compute/core/KernelDescriptors.h | |
parent | 22dd8b9014112fe446cb8cff6d52933d2603a97f (diff) | |
download | ComputeLibrary-2cc50b39613cea5e55c8a4851ee08d284a3d4f66.tar.gz |
Extended direct conv 2d interface for tuning the OpenCl kernel
Resolves COMPMID-5298
Change-Id: Ie9b907e5dcf86aa6add8d08799fa7ba7c264edea
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7888
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/KernelDescriptors.h')
-rw-r--r-- | arm_compute/core/KernelDescriptors.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/arm_compute/core/KernelDescriptors.h b/arm_compute/core/KernelDescriptors.h index b1086494e4..c45be9c06f 100644 --- a/arm_compute/core/KernelDescriptors.h +++ b/arm_compute/core/KernelDescriptors.h @@ -109,6 +109,15 @@ struct DWCComputeKernelInfo bool export_weights_to_cl_image{ false }; /**< Export the weights to cl_image */ }; +/** Compute descriptor used by the direct convolution kernel */ +struct DirectConvComputeKernelInfo +{ + int32_t m0{ 1 }; /**< Number of rows to be processed by the kernel */ + int32_t n0{ 1 }; /**< Number of columns to be processed by the kernel */ + int32_t k0{ 1 }; /**< Number of partial accumulations to be processed in a single iteration by the kernel */ + bool export_weights_to_cl_image{ false }; /**< Flag to export the weights to cl_image */ +}; + /** Descriptor used by the softmax kernels */ struct SoftmaxKernelInfo { |