diff options
Diffstat (limited to 'src/backends/neon/NeonLayerSupport.hpp')
| -rw-r--r-- | src/backends/neon/NeonLayerSupport.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/backends/neon/NeonLayerSupport.hpp b/src/backends/neon/NeonLayerSupport.hpp index b82351872b..783e6a0c46 100644 --- a/src/backends/neon/NeonLayerSupport.hpp +++ b/src/backends/neon/NeonLayerSupport.hpp @@ -246,7 +246,10 @@ public: const TensorInfo& output, const Pooling2dDescriptor& descriptor, Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override; - + bool IsPooling3dSupported(const TensorInfo& input, + const TensorInfo& output, + const Pooling3dDescriptor& descriptor, + Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override; bool IsPreluSupported(const TensorInfo& input, const TensorInfo& alpha, const TensorInfo& output, |