diff options
Diffstat (limited to 'src/armnn/LayerSupport.cpp')
-rw-r--r-- | src/armnn/LayerSupport.cpp | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/src/armnn/LayerSupport.cpp b/src/armnn/LayerSupport.cpp index fb3ce43646..5d2d205534 100644 --- a/src/armnn/LayerSupport.cpp +++ b/src/armnn/LayerSupport.cpp @@ -12,6 +12,7 @@ #include <cstring> #include <algorithm> #include <unordered_map> +#include <armnn/ArmNN.hpp> namespace armnn { @@ -100,6 +101,20 @@ bool IsBatchNormalizationSupported(const BackendId& backend, descriptor); } +bool IsBatchToSpaceNdSupported(const BackendId& backend, + const TensorInfo& input, + const TensorInfo& output, + const BatchToSpaceNdDescriptor& descriptor, + char* reasonIfUnsupported, + size_t reasonIfUnsupportedMaxLength) +{ + FORWARD_LAYER_SUPPORT_FUNC(backend, + IsBatchToSpaceNdSupported, + input, + output, + descriptor); +} + bool IsConstantSupported(const BackendId& backend, const TensorInfo& output, char* reasonIfUnsupported, |