From 21bda1405d2cb49fc873583b41a48836b33d285e Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Wed, 13 Mar 2024 16:10:32 +0000 Subject: IVGCVSW-8235 ScatterNd Operator Implementation (CL) Signed-off-by: Teresa Charlin Change-Id: I59fe96b0a272fa6984bfc172bf3e110476f3ce7b --- src/backends/cl/ClLayerSupport.hpp | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'src/backends/cl/ClLayerSupport.hpp') diff --git a/src/backends/cl/ClLayerSupport.hpp b/src/backends/cl/ClLayerSupport.hpp index 907db01b89..8e9c0be7f8 100644 --- a/src/backends/cl/ClLayerSupport.hpp +++ b/src/backends/cl/ClLayerSupport.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017-2023 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017-2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -300,6 +300,13 @@ public: const TensorInfo& output, Optional reasonIfUnsupported) const; + bool IsScatterNdSupported(const TensorInfo& input, + const TensorInfo& indices, + const TensorInfo& updates, + const TensorInfo& output, + const ScatterNdDescriptor& descriptor, + Optional reasonIfUnsupported = EmptyOptional()) const; + bool IsSliceSupported(const TensorInfo& input, const TensorInfo& output, const SliceDescriptor& descriptor, -- cgit v1.2.1