aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonLayerSupport.hpp
diff options
context:
space:
mode:
authorÉanna Ó Catháin <eanna.ocathain@arm.com>2019-01-25 10:01:40 +0000
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-01-25 14:51:53 +0000
commit12055747d47657a89d60748a078897f6436e6aa0 (patch)
tree0ac15d4e321ec5e2eaf0cbc074ac95c061289798 /src/backends/neon/NeonLayerSupport.hpp
parente9e7bfd09e3435d2bbb334e07c6a0a2514c80048 (diff)
downloadarmnn-12055747d47657a89d60748a078897f6436e6aa0.tar.gz
MLCE-84 Add Neon Pad support and unit tests
Change-Id: I0d949a9f23a61af5013efdd18572b29fae585f2a
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 631632485c..6de5c784cc 100644
--- a/src/backends/neon/NeonLayerSupport.hpp
+++ b/src/backends/neon/NeonLayerSupport.hpp
@@ -112,6 +112,11 @@ public:
bool IsOutputSupported(const TensorInfo& output,
Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
+ bool IsPadSupported(const TensorInfo& input,
+ const TensorInfo& output,
+ const PadDescriptor& descriptor,
+ Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
+
bool IsPermuteSupported(const TensorInfo& input,
const TensorInfo& output,
const PermuteDescriptor& descriptor,