diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-02-13 12:15:13 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:47:18 +0000 |
| commit | 284cfe2e3a44e5b20978e561c96c94d2193e93a1 (patch) | |
| tree | 204cb044578d66c89b3a60d0a3c8c7920c8a768e /arm_compute/runtime/NEON/functions/NEConvolutionLayer.h | |
| parent | f29975848a384fc127cf5401683fc246bab0d903 (diff) | |
| download | ComputeLibrary-284cfe2e3a44e5b20978e561c96c94d2193e93a1.tar.gz | |
COMPMID-903: Implements NEPermute for NHWC conversions
Change-Id: I4083e8d16bb23933634f229a1408dfd0e8f2922a
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/120069
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEConvolutionLayer.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEConvolutionLayer.h | 8 |
1 files changed, 8 insertions, 0 deletions
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; |