From 4fb5670e579f9c0e1403a8fe179a11a2c9f3a02f Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Wed, 10 Nov 2021 11:18:50 +0000 Subject: Rework gemm_reshape_rhs_(nt,t) with new macros Resolves COMPMID-4891 Change-Id: Ifdf2a0eaed23347a1b4465ea8d58c11b72083952 Signed-off-by: Gian Marco Iodice Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6741 Tested-by: Arm Jenkins Reviewed-by: Giorgio Arena --- src/core/CL/ICLKernel.h | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) (limited to 'src/core/CL/ICLKernel.h') diff --git a/src/core/CL/ICLKernel.h b/src/core/CL/ICLKernel.h index a7c979ef45..bc138e7e3f 100644 --- a/src/core/CL/ICLKernel.h +++ b/src/core/CL/ICLKernel.h @@ -226,6 +226,23 @@ public: add_tensor_argument<4>(idx, tensor, window); } + /** Add the passed NHW 3D tensor's parameters to the object's kernel's arguments by passing strides, dimensions and the offset to the first valid element in bytes. + * + * @param[in,out] idx Index at which to start adding the tensor's arguments. Will be incremented by the number of kernel arguments set. + * @param[in] tensor Tensor to set as an argument of the object's kernel. + */ + void add_3d_tensor_nhw_argument(unsigned int &idx, const ICLTensor *tensor); + + /** Returns the number of arguments enqueued per NHW 3D Tensor object. + * + * @return The number of arguments enqueued per NHW 3D Tensor object. + */ + constexpr static unsigned int num_arguments_per_3d_tensor_nhw() + { + constexpr unsigned int no_args_per_3d_tensor_nhw = 7u; + return no_args_per_3d_tensor_nhw; + } + /** Add the passed NHWC 4D tensor's parameters to the object's kernel's arguments by passing strides, dimensions and the offset to the first valid element in bytes. * * @param[in,out] idx Index at which to start adding the tensor's arguments. Will be incremented by the number of kernel arguments set. -- cgit v1.2.1