aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/LayerSupport.cpp
diff options
context:
space:
mode:
authorKevin May <kevin.may@arm.com>2019-12-12 17:28:05 +0000
committerKevin May <kevin.may@arm.com>2019-12-12 19:17:17 +0000
commit8ac5fc39b2556fd3ac0d0c43ebf55525bde97f58 (patch)
tree8fb4338647ef03e0dcbba3b6106967870f4d70c4 /src/armnn/LayerSupport.cpp
parente69fcc9ec42788218ba9693e53834b4606cf3882 (diff)
downloadarmnn-8ac5fc39b2556fd3ac0d0c43ebf55525bde97f58.tar.gz
IVGCVSW-4262 Use ACL Permute and Reshape Validate function in Neon and CL
!android-nn-driver:2465 Signed-off-by: Kevin May <kevin.may@arm.com> Change-Id: Ibabb73c0ae0df2e530a68398f75c76e6b80c0701
Diffstat (limited to 'src/armnn/LayerSupport.cpp')
-rw-r--r--src/armnn/LayerSupport.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/LayerSupport.cpp b/src/armnn/LayerSupport.cpp
index d730205981..38cac926de 100644
--- a/src/armnn/LayerSupport.cpp
+++ b/src/armnn/LayerSupport.cpp
@@ -510,11 +510,12 @@ bool IsPreluSupported(const BackendId& backend,
bool IsReshapeSupported(const BackendId& backend,
const TensorInfo& input,
+ const TensorInfo& output,
const ReshapeDescriptor& descriptor,
char* reasonIfUnsupported,
size_t reasonIfUnsupportedMaxLength)
{
- FORWARD_LAYER_SUPPORT_FUNC(backend, IsReshapeSupported, input, descriptor);
+ FORWARD_LAYER_SUPPORT_FUNC(backend, IsReshapeSupported, input, output, descriptor);
}
bool IsResizeSupported(const BackendId& backend,