aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKeith Davis <keith.davis@arm.com>2019-06-26 15:28:43 +0100
committerKeith Davis Arm <keith.davis@arm.com>2019-06-26 14:33:50 +0000
commit5191033c76330509666e0993857b7286e2a325fc (patch)
treee7b1222d249735ca4b874adb87ef799c5e2715bd
parente24e3cd94d0f5d6d8dbd5fec3e2d1e9385c6ff81 (diff)
downloadarmnn-5191033c76330509666e0993857b7286e2a325fc.tar.gz
IVGCVSW-3197 Support SPACE_TO_DEPTH on Android
* Fix for undefined reference to IsLayerSupported Signed-off-by: Keith Davis <keith.davis@arm.com> Change-Id: If9d39149a5ad41dbf835d79b8cb937ac253c927e
-rw-r--r--src/armnn/LayerSupport.cpp10
-rw-r--r--src/backends/reference/RefLayerSupport.cpp6
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);