From 284cfe2e3a44e5b20978e561c96c94d2193e93a1 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Tue, 13 Feb 2018 12:15:13 +0000 Subject: COMPMID-903: Implements NEPermute for NHWC conversions Change-Id: I4083e8d16bb23933634f229a1408dfd0e8f2922a Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/120069 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- arm_compute/runtime/NEON/functions/NEConvolutionLayer.h | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'arm_compute/runtime/NEON/functions/NEConvolutionLayer.h') diff --git a/arm_compute/runtime/NEON/functions/NEConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEConvolutionLayer.h index f65e7ef36b..f80f67d944 100644 --- a/arm_compute/runtime/NEON/functions/NEConvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEConvolutionLayer.h @@ -146,6 +146,14 @@ private: * except for input of QASYMM8 type where output should be of S32 type. */ void configure_mm(const ITensor *input, const ITensor *weights, ITensor *output); + /** Prepare the appropriate assembly optimized kernel + * + * @param[in] ci CPU information + * @param[in] M M parameter of matrix multiplication + * @param[in] N N parameter of matrix multiplication + * @param[in] K K parameter of matrix multiplication + */ + void configure_asm_mm(const struct CPUInfo &ci, int M, int N, int K); private: MemoryGroup _memory_group; -- cgit v1.2.1