diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2024-03-13 16:10:32 +0000 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2024-05-08 14:22:08 +0000 |
| commit | 21bda1405d2cb49fc873583b41a48836b33d285e (patch) | |
| tree | 6d57debcf6be6aeb28a3e3951757c73ae5636250 /src/backends/cl/ClLayerSupport.hpp | |
| parent | 8208e2b8b1d09d0e89394ae134eb61e390dfd93c (diff) | |
| download | armnn-21bda1405d2cb49fc873583b41a48836b33d285e.tar.gz | |
IVGCVSW-8235 ScatterNd Operator Implementation (CL)
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I59fe96b0a272fa6984bfc172bf3e110476f3ce7b
Diffstat (limited to 'src/backends/cl/ClLayerSupport.hpp')
| -rw-r--r-- | src/backends/cl/ClLayerSupport.hpp | 9 |
1 files changed, 8 insertions, 1 deletions
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<std::string&> reasonIfUnsupported) const; + bool IsScatterNdSupported(const TensorInfo& input, + const TensorInfo& indices, + const TensorInfo& updates, + const TensorInfo& output, + const ScatterNdDescriptor& descriptor, + Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const; + bool IsSliceSupported(const TensorInfo& input, const TensorInfo& output, const SliceDescriptor& descriptor, |