aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/NEON/functions/NEChannelShuffleLayer.h
diff options
context:
space:
mode:
authorMichele Di Giorgio <michele.digiorgio@arm.com>2021-09-03 10:29:49 +0100
committerMichele Di Giorgio <michele.digiorgio@arm.com>2021-09-03 13:16:52 +0000
commitb8025b3bb1b75fa94400a665e65a1d53ba9965f9 (patch)
tree733deafea2c0947d353345cef3e0835f1b149fb7 /arm_compute/runtime/NEON/functions/NEChannelShuffleLayer.h
parentaed61f2a14199fd54b8f7b0af9980689921fc0cc (diff)
downloadComputeLibrary-b8025b3bb1b75fa94400a665e65a1d53ba9965f9.tar.gz
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 <michele.digiorgio@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6204 Reviewed-by: Giorgio Arena <giorgio.arena@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEChannelShuffleLayer.h')
-rw-r--r--arm_compute/runtime/NEON/functions/NEChannelShuffleLayer.h1
1 files changed, 1 insertions, 0 deletions
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 |