aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonLayerSupport.hpp
diff options
context:
space:
mode:
authorPablo Tello <pablo.tello@arm.com>2020-01-23 10:37:17 +0000
committerPablo Tello <pablo.tello@arm.com>2020-01-24 14:50:49 +0000
commite61f0712dc480f70372fcfe0921a85a7de53661b (patch)
treea6a87887b8269a104b898a226c2d7cc31433de21 /src/backends/neon/NeonLayerSupport.hpp
parent0be3a88518a73819aaf71ef71f4fb2fbfc14bc88 (diff)
downloadarmnn-e61f0712dc480f70372fcfe0921a85a7de53661b.tar.gz
MLCE-156: Add Division workload in CpuAcc
Change-Id: I1f228fcaf1077867d9755a2b850c6703387fab34 Signed-off-by: Pablo Tello <pablo.tello@arm.com>
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 e49e78b3d2..f1d87f65f3 100644
--- a/src/backends/neon/NeonLayerSupport.hpp
+++ b/src/backends/neon/NeonLayerSupport.hpp
@@ -176,6 +176,11 @@ public:
const TensorInfo& output,
Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
+ bool IsDivisionSupported(const TensorInfo& input0,
+ const TensorInfo& input1,
+ const TensorInfo& output,
+ Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
+
bool IsNormalizationSupported(const TensorInfo& input,
const TensorInfo& output,
const NormalizationDescriptor& descriptor,