From b8025b3bb1b75fa94400a665e65a1d53ba9965f9 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Fri, 3 Sep 2021 10:29:49 +0100 Subject: List OpenCL kernels in Android.bp script Also add NHWC as supported data layout for the ChannelShuffle operator. Change-Id: I8487d0dfd1fdbc6350ef8b902a23fac6ac83407d Signed-off-by: Michele Di Giorgio Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6204 Reviewed-by: Giorgio Arena Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins --- arm_compute/runtime/CL/functions/CLChannelShuffleLayer.h | 1 + arm_compute/runtime/NEON/functions/NEChannelShuffleLayer.h | 1 + 2 files changed, 2 insertions(+) (limited to 'arm_compute') diff --git a/arm_compute/runtime/CL/functions/CLChannelShuffleLayer.h b/arm_compute/runtime/CL/functions/CLChannelShuffleLayer.h index d60548d9cc..3dc62595d2 100644 --- a/arm_compute/runtime/CL/functions/CLChannelShuffleLayer.h +++ b/arm_compute/runtime/CL/functions/CLChannelShuffleLayer.h @@ -46,6 +46,7 @@ public: * * Valid data layouts: * - NCHW + * - NHWC * * Valid data type configurations: * |src |dst | diff --git a/arm_compute/runtime/NEON/functions/NEChannelShuffleLayer.h b/arm_compute/runtime/NEON/functions/NEChannelShuffleLayer.h index 8888efec4f..bc19e1a4af 100644 --- a/arm_compute/runtime/NEON/functions/NEChannelShuffleLayer.h +++ b/arm_compute/runtime/NEON/functions/NEChannelShuffleLayer.h @@ -46,6 +46,7 @@ public: * * Valid data layouts: * - NCHW + * - NHWC * * Valid data type configurations: * |src |dst | -- cgit v1.2.1