diff options
| -rw-r--r-- | src/armnn/LayerSupport.cpp | 10 | ||||
| -rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 6 |
2 files changed, 13 insertions, 3 deletions
diff --git a/src/armnn/LayerSupport.cpp b/src/armnn/LayerSupport.cpp index 1d9d249024..3c3b9e352e 100644 --- a/src/armnn/LayerSupport.cpp +++ b/src/armnn/LayerSupport.cpp @@ -535,6 +535,16 @@ bool IsSpaceToBatchNdSupported(const BackendId& backend, FORWARD_LAYER_SUPPORT_FUNC(backend, IsSpaceToBatchNdSupported, input, output, descriptor); } +bool IsSpaceToDepthSupported(const BackendId& backend, + const TensorInfo& input, + const TensorInfo& output, + const SpaceToDepthDescriptor& descriptor, + char* reasonIfUnsupported, + size_t reasonIfUnsupportedMaxLength) +{ + FORWARD_LAYER_SUPPORT_FUNC(backend, IsSpaceToDepthSupported, input, output, descriptor); +} + ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead") bool IsSplitterSupported(const BackendId& backend, const TensorInfo& input, diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 1f37420e42..06d9e1bff9 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -1337,9 +1337,9 @@ bool RefLayerSupport::IsSpaceToBatchNdSupported(const TensorInfo& input, } bool RefLayerSupport::IsSpaceToDepthSupported(const TensorInfo& input, - const TensorInfo& output, - const SpaceToDepthDescriptor& descriptor, - Optional<std::string&> reasonIfUnsupported) const + const TensorInfo& output, + const SpaceToDepthDescriptor& descriptor, + Optional<std::string&> reasonIfUnsupported) const { ignore_unused(descriptor); |