From f40d6d41b58a086273513d1b386725a7ae9c5503 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Thu, 22 Apr 2021 09:12:11 +0100 Subject: IVGCVSW-5420 'Add CL support for CAST' Signed-off-by: Sadik Armagan Change-Id: I1bbf1f2edf7282cb69b99e22de1b8a2fe6e443c2 --- src/backends/cl/ClLayerSupport.hpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'src/backends/cl/ClLayerSupport.hpp') diff --git a/src/backends/cl/ClLayerSupport.hpp b/src/backends/cl/ClLayerSupport.hpp index 8b873915dd..b7815416f4 100644 --- a/src/backends/cl/ClLayerSupport.hpp +++ b/src/backends/cl/ClLayerSupport.hpp @@ -52,6 +52,10 @@ public: const BatchToSpaceNdDescriptor& descriptor, Optional reasonIfUnsupported = EmptyOptional()) const override; + bool IsCastSupported(const TensorInfo& input, + const TensorInfo& output, + Optional reasonIfUnsupported = EmptyOptional()) const override; + bool IsComparisonSupported(const TensorInfo& input0, const TensorInfo& input1, const TensorInfo& ouput, -- cgit v1.2.1