aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonLayerSupport.hpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2020-04-10 19:24:55 +0100
committerTeresaARM <teresa.charlinreyes@arm.com>2020-05-29 16:21:58 +0000
commitf540eb8111ce5d241111da487be7d817661e29b4 (patch)
tree6a22c7f0076cce9a0e304120d5b1f4c0d1b4cfc0 /src/backends/neon/NeonLayerSupport.hpp
parent3940d8bfcaad963e95b882984a93ccd69455822a (diff)
downloadarmnn-f540eb8111ce5d241111da487be7d817661e29b4.tar.gz
IVGCVSW-3846 Add NEON GATHER Workload
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: I1a66fdad63cef16866d9dfcb8a339647f856e1d4
Diffstat (limited to 'src/backends/neon/NeonLayerSupport.hpp')
-rw-r--r--src/backends/neon/NeonLayerSupport.hpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/neon/NeonLayerSupport.hpp b/src/backends/neon/NeonLayerSupport.hpp
index adb1891de8..25817822fd 100644
--- a/src/backends/neon/NeonLayerSupport.hpp
+++ b/src/backends/neon/NeonLayerSupport.hpp
@@ -123,6 +123,11 @@ public:
const FullyConnectedDescriptor& descriptor,
Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
+ bool IsGatherSupported(const TensorInfo& input0,
+ const TensorInfo& input1,
+ const TensorInfo& output,
+ Optional<std::string&> reasonIfUnsupported) const override;
+
ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
bool IsGreaterSupported(const TensorInfo& input0,
const TensorInfo& input1,