From 1222dbd3ddc4e819f8d1b204535bfd222cd1eadd Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Thu, 2 Sep 2021 13:58:52 +0100 Subject: IVGCVSW-3706 Add Channel Shuffle Workload to GpuAcc backend Signed-off-by: Teresa Charlin Change-Id: Ic37fc49a97a5ca570a23bc415d15ee3841534336 --- src/backends/cl/ClLayerSupport.hpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/backends/cl/ClLayerSupport.hpp') diff --git a/src/backends/cl/ClLayerSupport.hpp b/src/backends/cl/ClLayerSupport.hpp index b7815416f4..43ae428163 100644 --- a/src/backends/cl/ClLayerSupport.hpp +++ b/src/backends/cl/ClLayerSupport.hpp @@ -56,6 +56,11 @@ public: const TensorInfo& output, Optional reasonIfUnsupported = EmptyOptional()) const override; + bool IsChannelShuffleSupported(const TensorInfo& input, + const TensorInfo& output, + const ChannelShuffleDescriptor& descriptor, + Optional reasonIfUnsupported = EmptyOptional()) const override; + bool IsComparisonSupported(const TensorInfo& input0, const TensorInfo& input1, const TensorInfo& ouput, -- cgit v1.2.1